wip

KotlinIsland · Aug 29, 2024 · f77242c · f77242c
1 parent 3055bb2
commit f77242c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/basedtyping/transformer.py b/src/basedtyping/transformer.py
@@ -122,7 +122,9 @@ def visit_Subscript(self, node: ast.Subscript) -> ast.AST:
                 if isinstance(slice_, ast.Tuple):
                     slice_.elts[0] = self.visit(slice_.elts[0])
                 else:
-                    node.slice = self.visit(slice_)
+                    temp = self.visit(slice_)
+                    assert isinstance(temp, ast.expr)
+                    node.slice = temp
             return node
         with self.implicit_tuple():
             result = self.generic_visit(node)
@@ -220,6 +222,7 @@ def crifigy_type(
         *,
         string_literals: bool,
     ) -> object:
+        tree: ast.AST
         try:
             tree = ast.parse(value, mode="eval")
         except SyntaxError:

diff --git a/tests/test_is_subform.py b/tests/test_is_subform.py
@@ -26,6 +26,6 @@ def test_old_union():
     assert issubform(Union[int, str], Union[str, int])
     if sys.version_info >= (3, 10):
         assert issubform(
-            Union[int, str],  # type: ignore[arg-type]
+            Union[int, str],
             int | str,  # type: ignore[unused-ignore, arg-type]
         )