From f4cf6b6f4addbdec499e225ffe8916100b1c4b6f Mon Sep 17 00:00:00 2001 From: hauntsaninja Date: Thu, 17 Oct 2024 01:30:17 -0700 Subject: [PATCH] [mypyc] fix relative imports in __init__.py Fixes https://github.com/mypyc/mypyc/issues/1070 --- mypyc/irbuild/statement.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mypyc/irbuild/statement.py b/mypyc/irbuild/statement.py index 4d828b1b9d82..ac3fa33f36ca 100644 --- 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 = ""