Change typing from List[int] to npt.ArrayLike

jni · Oct 16, 2023 · 0241322 · 0241322
1 parent c7fe159
commit 0241322
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/skan/csr.py b/src/skan/csr.py
@@ -6,7 +6,7 @@
 from skimage import morphology
 from skimage.graph import central_pixel
 from skimage.util._map_array import map_array, ArrayMap
-from typing import List
+import numpy.typing as npt
 import numba
 import warnings
 
@@ -656,7 +656,7 @@ def path_stdev(self):
         means = self.path_means()
         return np.sqrt(np.clip(sumsq/lengths - means*means, 0, None))
 
-    def prune_paths(self, indices: List[int]) -> 'Skeleton':
+    def prune_paths(self, indices: npt.ArrayLike) -> 'Skeleton':
         """Prune nodes from the skeleton.
 
         Parameters