Fix typo in prediction script

computational-cell-analytics · Jul 11, 2024 · 4703502 · 4703502
1 parent 826bbe4
commit 4703502
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/prediction/run_prediction_distance_unet.py b/scripts/prediction/run_prediction_distance_unet.py
@@ -32,7 +32,7 @@ def main():
         block_shape = (64, 256, 256) if have_cuda else (64, 64, 64)
         halo = (16, 64, 64) if have_cuda else (8, 32, 32)
     else:
-        chunks = z5py.File(args.input_path, "r")[args.input_key].chunks
+        chunks = z5py.File(args.input, "r")[args.input_key].chunks
         block_shape = tuple([2 * ch for ch in chunks]) if have_cuda else tuple(chunks)
         halo = (16, 64, 64) if have_cuda else (8, 32, 32)