fixed import TypeAlias for python < 3.10

leon-vv · Jan 23, 2025 · 23f7047 · 23f7047
1 parent bddede1
commit 23f7047
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 7 deletions.
diff --git a/traceon/_typing/_array_like.py b/traceon/_typing/_array_like.py
@@ -1,7 +1,11 @@
 from __future__ import annotations
-
+import sys
 from collections.abc import Sequence
-from typing import TypeAlias
+
+if sys.version_info >= (3, 10):
+    from typing import TypeAlias
+else:
+    from typing_extensions import TypeAlias
 
 import numpy as np
 

diff --git a/traceon/_typing/_backend.py b/traceon/_typing/_backend.py
@@ -1,7 +1,13 @@
 from __future__ import annotations
-
-from typing import Literal, TypeAlias
+import sys
+from typing import Literal
 from collections.abc import Sequence
+
+if sys.version_info >= (3, 10):
+    from typing import TypeAlias
+else:
+    from typing_extensions import TypeAlias
+
 import numpy as np
 
 from ._array_like import _NumpyFloat

diff --git a/traceon/_typing/_geometric_types.py b/traceon/_typing/_geometric_types.py
@@ -1,6 +1,13 @@
 from __future__ import annotations
-from typing import Literal, TypeAlias
+import sys
+from typing import Literal
 from collections.abc import Callable, Sequence
+
+if sys.version_info >= (3, 10):
+    from typing import TypeAlias
+else:
+    from typing_extensions import TypeAlias
+
 import numpy as np
 from ._array_like import _NumpyFloat
 

diff --git a/traceon/_typing/_mesh_types.py b/traceon/_typing/_mesh_types.py
@@ -1,7 +1,13 @@
 from __future__ import annotations
-
-from typing import Literal, TypeAlias
+import sys
+from typing import Literal
 from collections.abc import Sequence, Mapping
+
+if sys.version_info >= (3, 10):
+    from typing import TypeAlias
+else:
+    from typing_extensions import TypeAlias
+
 import numpy as np
 
 from ._array_like import _NumpyFloat, _NumpyInt, ArrayInt1D