Updating tests for SupervisedTrainer default change.

id-b3 · id-b3 · commit 4afd580dfbf4 · 2025-08-19T09:50:43.000Z
diff --git a/tests/apps/deepgrow/transforms/test_deepgrow_interaction.py b/tests/apps/deepgrow/transforms/test_deepgrow_interaction.py
@@ -78,6 +78,7 @@ def run_interaction(self, train, compose):
             optimizer=opt,
             loss_function=loss,
             iteration_update=i,
+            decollate=True,
         )
         engine.add_event_handler(IterationEvents.INNER_ITERATION_STARTED, add_one)
         engine.add_event_handler(IterationEvents.INNER_ITERATION_COMPLETED, add_one)
diff --git a/tests/integration/test_deepedit_interaction.py b/tests/integration/test_deepedit_interaction.py
@@ -103,6 +103,7 @@ def run_interaction(self, train):
             loss_function=loss,
             postprocessing=post_transforms,
             iteration_update=i,
+            decollate=True,
         )
         engine.add_event_handler(IterationEvents.INNER_ITERATION_STARTED, add_one)
         engine.add_event_handler(IterationEvents.INNER_ITERATION_COMPLETED, add_one)
diff --git a/tests/testing_data/config_fl_train.json b/tests/testing_data/config_fl_train.json
@@ -119,7 +119,8 @@
             "loss_function": "@loss",
             "optimizer": "@optimizer",
             "inferer": "@train#inferer",
-            "train_handlers": "@train#handlers"
+            "train_handlers": "@train#handlers",
+            "decollate": true
         }
     },
     "validate": {

Original file line number	Diff line number	Diff line change
`@@ -78,6 +78,7 @@ def run_interaction(self, train, compose):`
`78`	`78`	`optimizer=opt,`
`79`	`79`	`loss_function=loss,`
`80`	`80`	`iteration_update=i,`
	`81`	`+ decollate=True,`
`81`	`82`	`)`
`82`	`83`	`engine.add_event_handler(IterationEvents.INNER_ITERATION_STARTED, add_one)`
`83`	`84`	`engine.add_event_handler(IterationEvents.INNER_ITERATION_COMPLETED, add_one)`
Original file line number	Diff line number	Diff line change
`@@ -103,6 +103,7 @@ def run_interaction(self, train):`
`103`	`103`	`loss_function=loss,`
`104`	`104`	`postprocessing=post_transforms,`
`105`	`105`	`iteration_update=i,`
	`106`	`+ decollate=True,`
`106`	`107`	`)`
`107`	`108`	`engine.add_event_handler(IterationEvents.INNER_ITERATION_STARTED, add_one)`
`108`	`109`	`engine.add_event_handler(IterationEvents.INNER_ITERATION_COMPLETED, add_one)`