From addff03b1a127bec0240e6582d8e7660860f3437 Mon Sep 17 00:00:00 2001
From: Xavier Olive <git@xoolive.org>
Date: Mon, 18 Nov 2024 22:06:18 +0100
Subject: [PATCH] fix some typing

---
 pyproject.toml          | 2 +-
 src/impunity/visitor.py | 4 ++--
 tests/sample_module.py  | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 93c5681..024f3d9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,7 +55,7 @@ known-first-party = ["numpy", "pandas"]
 source = ["src"]
 
 [tool.mypy]
-python_version = 3.8
+python_version = "3.9"
 platform = "posix"
 
 color_output = true
diff --git a/src/impunity/visitor.py b/src/impunity/visitor.py
index cd950b4..fa3f6a6 100644
--- a/src/impunity/visitor.py
+++ b/src/impunity/visitor.py
@@ -148,8 +148,8 @@ def get_annotation_unit(self, node: ast.expr) -> Optional[str]:
 
         elif isinstance(node, ast.Subscript):
             if isinstance(node.slice, ast.Index):
-                if isinstance(node.slice.value, ast.Tuple):
-                    unit_node = node.slice.value.elts[1]
+                if isinstance(node.slice.value, ast.Tuple):  # type: ignore
+                    unit_node = node.slice.value.elts[1]  # type: ignore
             elif isinstance(node.slice, ast.Tuple):
                 unit_node = node.slice.elts[1]
             if isinstance(unit_node, ast.Constant):
diff --git a/tests/sample_module.py b/tests/sample_module.py
index bea68fb..730a9f4 100644
--- a/tests/sample_module.py
+++ b/tests/sample_module.py
@@ -1,12 +1,12 @@
 from typing import Any
 
-from typing_extensions import Annotated
+from typing_extensions import Annotated, TypeAlias
 
 import numpy as np
 import numpy.typing as npt
 from impunity import impunity
 
-NDArrayFloat = npt.NDArray[np.float64]
+NDArrayFloat: TypeAlias = npt.NDArray[np.float64]
 meters = Annotated[NDArrayFloat, "m"]
 seconds = Annotated[float, "s"]
 meters_per_second = Annotated[NDArrayFloat, "m/s"]