fix: no reference to sharding in the unsharded function

seung-lab · Oct 3, 2023 · ee15a88 · ee15a88
1 parent 66f86b3
commit ee15a88
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/igneous/task_creation/image.py b/igneous/task_creation/image.py
@@ -235,6 +235,7 @@ def create_downsampling_tasks(
   memory_target: size in bytes to target for memory usage
   """
   def ds_shape(mip, chunk_size=None, factor=None):
+    nonlocal num_mips
     if chunk_size:
       shape = Vec(*chunk_size)
     else:
@@ -243,9 +244,7 @@ def ds_shape(mip, chunk_size=None, factor=None):
     if factor is None:
       factor = downsample_scales.axis_to_factor(axis)
 
-    if sharding:
-      num_mips = 1
-    elif num_mips is None:
+    if num_mips is None:
       num_mips = num_mips_from_memory_target(
         memory_target, vol.dtype, shape, factor
       )

diff --git a/igneous_cli/cli.py b/igneous_cli/cli.py
@@ -135,7 +135,7 @@ def compute_bounds(path, mip, xrange, yrange, zrange):
 
 @click.group()
 @click.option("-p", "--parallel", default=1, help="Run with this number of parallel processes. If 0, use number of cores.")
-@click.version_option(version="4.19.3")
+@click.version_option(version="4.20.0")
 @click.pass_context
 def main(ctx, parallel):
   """