fix: fix top-level key typo in predictor

okunator · okunator · commit 624328218aef · 2025-07-25T11:25:58.000+03:00
diff --git a/cellseg_models_pytorch/inference/predictor.py b/cellseg_models_pytorch/inference/predictor.py
@@ -451,7 +451,7 @@ def _predict_sliding_win(
         soft_masks["nuc"].type_map = (nuc / recovery)[..., pady:-pady, padx:-padx]
         soft_masks["nuc"].aux_map = (nuc_aux / recovery)[..., pady:-pady, padx:-padx]
         if nuc_binary is not None:
-            soft_masks["binary_map"].aux_map = (nuc_binary / recovery)[
+            soft_masks["nuc"].aux_map = (nuc_binary / recovery)[
                 ..., pady:-pady, padx:-padx
             ]
 

Original file line number	Diff line number	Diff line change
`@@ -451,7 +451,7 @@ def _predict_sliding_win(`
`451`	`451`	`soft_masks["nuc"].type_map = (nuc / recovery)[..., pady:-pady, padx:-padx]`
`452`	`452`	`soft_masks["nuc"].aux_map = (nuc_aux / recovery)[..., pady:-pady, padx:-padx]`
`453`	`453`	`if nuc_binary is not None:`
`454`		`- soft_masks["binary_map"].aux_map = (nuc_binary / recovery)[`
	`454`	`+ soft_masks["nuc"].aux_map = (nuc_binary / recovery)[`
`455`	`455`	`..., pady:-pady, padx:-padx`
`456`	`456`	`]`
`457`	`457`