Skip to content

Commit e42bd0b

Browse files
Fix critical bug in predict with tiling
1 parent cc6ff24 commit e42bd0b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

bioimageio/core/prediction.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,13 +330,16 @@ def check_tiling(tiling):
330330
# override metadata defaults with provided dict
331331
if isinstance(tiling, dict):
332332
for key in ["halo", "tile", "scale"]:
333-
default_tiling.update(tiling.get(key, dict()))
333+
default_tiling[key].update(tiling.get(key, dict()))
334334
tiling = default_tiling
335335
check_tiling(tiling)
336+
336337
elif isinstance(tiling, bool) and not tiling:
337338
raise NotImplementedError("Should be unreachable")
339+
338340
else:
339341
raise ValueError(f"Invalid argument for tiling: {tiling}")
342+
340343
return tiling
341344

342345

0 commit comments

Comments
 (0)