diff --git a/codeflash/languages/python/static_analysis/line_profile_utils.py b/codeflash/languages/python/static_analysis/line_profile_utils.py index 93997b2c6..f2a5da644 100644 --- a/codeflash/languages/python/static_analysis/line_profile_utils.py +++ b/codeflash/languages/python/static_analysis/line_profile_utils.py @@ -56,7 +56,6 @@ def visit_ImportFrom(self, node: ast.ImportFrom) -> None: local_name = alias.asname if alias.asname else alias.name # For from imports, we store (module_name, imported_name) self.import_aliases[local_name] = (node.module, alias.name) - self.generic_visit(node) def visit_FunctionDef(self, node: ast.FunctionDef) -> None: """Check function decorators for JIT decorators."""