Fix numpy

seung-lab · Nov 30, 2023 · b9ed8e7 · b9ed8e7
1 parent e5c023f
commit b9ed8e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/cut_chunk_common.py b/scripts/cut_chunk_common.py
@@ -42,8 +42,8 @@ def cut_data(data, start_coord, end_coord, padding):
     bb = tuple(slice(start_coord[i], end_coord[i]) for i in range(3))
     if data.shape[3] == 1:
         if data.dtype == 'float32':
-            pmap = np.squeeze(data[bb])
-            affinity = [np.minimum(np.roll(pmap, shift=1, axis=axis), pmap) for axis in range(3)]
+            pmap = numpy.squeeze(data[bb])
+            affinity = [numpy.minimum(numpy.roll(pmap, shift=1, axis=axis), pmap) for axis in range(3)]
             return pad_data(numpy.stack(affinity, axis=-1), padding)
         else:
             return pad_data(data[bb], padding)