Fix

neuro-ml · Dec 26, 2023 · b5c453a · b5c453a
1 parent 1b7578d
commit b5c453a
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/imops/interp2d.py b/imops/interp2d.py
@@ -58,12 +58,13 @@ def __init__(
 
         if not isinstance(points, np.ndarray):
             raise ValueError(f'Wrong type of `points` argument, expected np.ndarray. Got {type(points)}')
-
-        if not isinstance(self.values, np.ndarray):
-            raise ValueError(f'Wrong type of `values` argument, expected np.ndarray. Got {type(self.values)}')
 
-        if self.values.ndim > 1:
-            raise ValueError(f'Wrong shape of `values` argument, expected ndim=1. Got shape {self.values.shape}')
+        if values is not None:
+            if not isinstance(values, np.ndarray):
+                raise ValueError(f'Wrong type of `values` argument, expected np.ndarray. Got {type(values)}')
+
+            if values.ndim > 1:
+                raise ValueError(f'Wrong shape of `values` argument, expected ndim=1. Got shape {values.shape}')
 
         super().__init__(points, num_threads, triangles)
         self.kdtree = KDTree(data=points, **kwargs)