def METHOD_NAME(class_name):
    def remove_import_transformer(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)
    return remove_import_transformer