Skip to content

Commit 6418e7c

Browse files
2 parents 7f3b265 + d9626e4 commit 6418e7c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

flamingo_tools/segmentation/unet_prediction.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -575,6 +575,8 @@ def run_unet_prediction_slurm(
575575
"""
576576
os.makedirs(output_folder, exist_ok=True)
577577
prediction_instances = int(prediction_instances)
578+
if isinstance(scale, str):
579+
scale = float(scale)
578580
slurm_task_id = os.environ.get("SLURM_ARRAY_TASK_ID")
579581

580582
if s3 is not None:

0 commit comments

Comments
 (0)