extend with tuples

HugoMVale · Dec 3, 2023 · ddbcf07 · ddbcf07
1 parent 49fc823
commit ddbcf07
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/polykin/types.py b/src/polykin/types.py
@@ -7,17 +7,17 @@
 from nptyping import NDArray, Shape, Int32, Float64
 
 IntArray = NDArray[Any, Int32]
-IntArrayLike = Union[list[int], IntArray]
+IntArrayLike = Union[list[int], tuple[int, ...], IntArray]
 IntOrArray = Union[int, IntArray]
 IntOrArrayLike = Union[int, IntArrayLike]
 
 FloatArray = NDArray[Any, Float64]
-FloatArrayLike = Union[list[float], FloatArray]
+FloatArrayLike = Union[list[float], tuple[float, ...], FloatArray]
 FloatOrArray = Union[float, FloatArray]
 FloatOrArrayLike = Union[float, FloatArrayLike]
 
 FloatVector = NDArray[Shape['*'], Float64]
-FloatVectorLike = Union[list[float], FloatVector]
+FloatVectorLike = Union[list[float], tuple[float, ...], FloatVector]
 FloatOrVector = Union[float, FloatVector]
 FloatOrVectorLike = Union[float, FloatVectorLike]