add IntVector and IntVectorLike

HugoMVale · Feb 25, 2024 · 69228f9 · 69228f9
1 parent d6fc753
commit 69228f9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/polykin/utils/types.py b/src/polykin/utils/types.py
@@ -14,6 +14,9 @@
 IntOrArray = Union[int, IntArray]
 IntOrArrayLike = Union[int, IntArrayLike]
 
+IntVector = NDArray[Shape['*'], Int32]
+IntVectorLike = Union[list[int], tuple[int, ...], IntVector]
+
 FloatArray = NDArray[Any, Float64]
 FloatArrayLike = Union[list[float], tuple[float, ...], FloatArray]
 FloatOrArray = Union[float, FloatArray]