wip

KotlinIsland · Sep 5, 2024 · 61b6665 · 61b6665
1 parent aeef2fc
commit 61b6665
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 7 deletions.
diff --git a/src/basedtyping/__init__.py b/src/basedtyping/__init__.py
@@ -639,7 +639,30 @@ def __init__(self, arg: str, *, is_argument=True, module: object = None, is_clas
         self.__forward_is_class__ = is_class
         self.__forward_module__ = module
 
-    if sys.version_info >= (3, 9):
+    if sys.version_info >= (3, 13):
+        @typing.overload
+        def _evaluate(
+                self, globalns: dict[str, object] | None, localns: dict[str, object] | None, *, recursive_guard: frozenset[str]
+        ) -> object | None: ...
+        @typing.overload
+        def _evaluate(
+                self,
+                globalns: dict[str, object] | None,
+                localns: dict[str, object] | None,
+                type_params: tuple[TypeVar | typing.ParamSpec | typing.TypeVarTuple, ...],
+                *,
+                recursive_guard: frozenset[str],
+        ) -> object | None: ...
+    elif sys.version_info >= (3, 12):
+        def _evaluate(
+                self,
+                globalns: dict[str, object] | None,
+                localns: dict[str, object] | None,
+                type_params: tuple[TypeVar | typing.ParamSpec | typing.TypeVarTuple, ...] | None = None,
+                *,
+                recursive_guard: frozenset[str],
+        ) -> object | None: ...
+    elif sys.version_info >= (3, 9):
         def _evaluate(
                 self, globalns: dict[str, object] | None, localns: dict[str, object] | None, recursive_guard: frozenset[str]
         ) -> object | None:

diff --git a/src/basedtyping/transformer.py b/src/basedtyping/transformer.py
@@ -123,12 +123,9 @@ def visit_Subscript(self, node: ast.Subscript) -> ast.AST:
             else:
                 slice_ = node.slice
                 if isinstance(slice_, ast.Tuple):
-                    if sys.version_info < (3, 12):
-                        temp = self.visit(slice_.elts[0])
-                        assert isinstance(temp, ast.expr)
-                        slice_.elts[0] = temp
-                    else:
-                        slice_.elts[0] = self.visit(slice_.elts[0])
+                    temp = self.visit(slice_.elts[0])
+                    assert isinstance(temp, ast.expr)
+                    slice_.elts[0] = temp
                 else:
                     temp = self.visit(slice_)
                     assert isinstance(temp, ast.expr)