From 47035023fb0407450353ebb82212991004f93852 Mon Sep 17 00:00:00 2001 From: Constantin Pape Date: Thu, 11 Jul 2024 13:02:49 +0200 Subject: [PATCH] Fix typo in prediction script --- scripts/prediction/run_prediction_distance_unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/prediction/run_prediction_distance_unet.py b/scripts/prediction/run_prediction_distance_unet.py index 3ad5ddc..93bb32a 100644 --- 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)