def METHOD_NAME(node, capture, filename):
    if "class_import" in capture and capture["class_name"].value == class_name:
        if capture["class_import"].type == python_symbols.import_from:
            import_from_stmt = node.children
            imported_classes = import_from_stmt[3]
            if len(imported_classes.children) > 1:
                # something of the form `from feast import A, ValueType`
                for i, class_leaf in enumerate(imported_classes.children):
                    if class_leaf.value == class_name:
                        imported_classes.children.pop(i)
                        if i == len(imported_classes.children):
                            imported_classes.children.pop(i - 1)
                        else:
                            imported_classes.children.pop(i)
            else:
                # something of the form `from feast import ValueType`
                node.parent.children.remove(node)