refactor: added support for ArrayLike in ArrayLike1dType and ArrayLik…

…e2dType
siapy · Oct 1, 2024 · 368fdeb · 368fdeb
1 parent 826b05b
commit 368fdeb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/siapy/core/types.py b/siapy/core/types.py
@@ -3,6 +3,7 @@
 import numpy as np
 import pandas as pd
 import spectral as sp
+from numpy.typing import ArrayLike
 from PIL.Image import Image
 
 from siapy.entities import SpectralImage, SpectralImageSet
@@ -34,5 +35,5 @@
     | np.uint64
 )
 ImageContainerType = SpectralImage | SpectralImageSet
-ArrayLike1dType = np.ndarray | pd.Series | Sequence[Any]
-ArrayLike2dType = np.ndarray | pd.DataFrame | Sequence[Any]
+ArrayLike1dType = np.ndarray | pd.Series | Sequence[Any] | ArrayLike
+ArrayLike2dType = np.ndarray | pd.DataFrame | Sequence[Any] | ArrayLike