fix TYPE_CHECKING condition checking

reagento · Aug 29, 2024 · 26c8823 · 26c8823
1 parent 6b38a8c
commit 26c8823
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/adaptix/_internal/type_tools/type_evaler.py b/src/adaptix/_internal/type_tools/type_evaler.py
@@ -7,7 +7,11 @@
 def make_fragments_collector(*, typing_modules: Sequence[str]) -> Callable[[ast.Module], list[ast.stmt]]:
     def check_condition(expr: ast.expr) -> bool:
         # searches for `TYPE_CHECKING`
-        if isinstance(expr, ast.Name) and isinstance(expr.ctx, ast.Load):
+        if (
+            isinstance(expr, ast.Name)
+            and isinstance(expr.ctx, ast.Load)
+            and expr.id == "TYPE_CHECKING"
+        ):
             return True
 
         # searches for `typing.TYPE_CHECKING`