fixed issue with type subscription for python <3.9

leon-vv · Jan 23, 2025 · 5faf7dc · 5faf7dc
1 parent 23f7047
commit 5faf7dc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/traceon/_typing/_array_like.py b/traceon/_typing/_array_like.py
@@ -9,8 +9,12 @@
 
 import numpy as np
 
-_NumpyFloat: TypeAlias = np.dtype[np.floating]
-_NumpyInt: TypeAlias = np.dtype[np.integer]
+if sys.version_info >= (3, 9):
+    _NumpyFloat: TypeAlias = np.dtype[np.floating]
+    _NumpyInt: TypeAlias = np.dtype[np.integer]
+else:
+    _NumpyFloat: TypeAlias = np.dtype
+    _NumpyInt: TypeAlias = np.dtype
 
 _ShapeLike: TypeAlias = tuple[int, ...]