[mypyc] fix relative imports in __init__.py (#17979)

Fixes mypyc/mypyc#1070 Missed in #10543
python · Oct 21, 2024 · cb9dd7d · cb9dd7d
1 parent b16c192
commit cb9dd7d
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 = ""