Merge pull request #383 from deshaw/fix/auto-imports

Fix lazy loading of variables
deshaw · Dec 6, 2024 · ace50d6 · ace50d6
2 parents c73b22c + 84d40cb
commit ace50d6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/python/pyflyby/_dynimp.py b/lib/python/pyflyby/_dynimp.py
@@ -111,8 +111,9 @@ def _add_import(ip, names: str, code: str) -> None:
     private version of add_import
     """
     assert ip is not None
-    mang = PYFLYBY_LAZY_LOAD_PREFIX + names.replace(",", "_").replace(" ", "_")
-    a: FrozenSet[Import] = ImportSet(f"{mang} import {names}")._importset
+    module = PYFLYBY_LAZY_LOAD_PREFIX.split()[1]
+    mang = module + names.replace(",", "_").replace(" ", "_")
+    a: FrozenSet[Import] = ImportSet(f"from {mang} import {names}")._importset
     b: FrozenSet[Import] = ip._auto_importer.db.known_imports._importset
     s_import: FrozenSet[Import] = a | b