fixing chunk size for single planes

google · Feb 16, 2024 · 667d203 · 667d203
1 parent fcd47a6
commit 667d203
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/python/neuroglancer/write_annotations.py b/python/neuroglancer/write_annotations.py
@@ -521,12 +521,11 @@ def write(self, path: Union[str, pathlib.Path]):
         sharding_spec = choose_output_spec(len(self.annotations), total_ann_bytes)
 
         # calculate the number of chunks in each dimension
-        num_chunks = np.max(
-            np.full(self.upper_bound.shape, 1, dtype=int),
-            np.ceil((self.upper_bound - self.lower_bound) / self.chunk_size).astype(
-                int
-            ),
-        )
+        num_chunks = np.ceil(
+            (self.upper_bound - self.lower_bound) / self.chunk_size
+        ).astype(int)
+
+        np.maximum(num_chunks, np.full(num_chunks.shape, 1, dtype=int))
 
         # make directories
         os.makedirs(path, exist_ok=True)