[mypyc] fix relative imports in __init__.py

Fixes mypyc/mypyc#1070
hauntsaninja · Oct 17, 2024 · f4cf6b6 · f4cf6b6
1 parent c201a18
commit f4cf6b6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mypyc/irbuild/statement.py b/mypyc/irbuild/statement.py
@@ -347,10 +347,10 @@ def transform_import_from(builder: IRBuilder, node: ImportFrom) -> None:
         return
 
     module_state = builder.graph[builder.module_name]
-    if module_state.ancestors is not None and module_state.ancestors:
-        module_package = module_state.ancestors[0]
-    elif builder.module_path.endswith("__init__.py"):
+    if builder.module_path.endswith("__init__.py"):
         module_package = builder.module_name
+    elif module_state.ancestors is not None and module_state.ancestors:
+        module_package = module_state.ancestors[0]
     else:
         module_package = ""