add comment to test_fit_partial_multiple_times test

nsundalov · nsundalov · commit 78b65d72a5c5 · 2025-07-15T18:00:49.000+03:00
diff --git a/tests/models/nn/transformers/test_base.py b/tests/models/nn/transformers/test_base.py
@@ -483,6 +483,9 @@ def on_train_epoch_start(self) -> None:
         )
         t2 = deepcopy(model.fit_trainer)
 
+        # Since for the second we are fitting on an empty dataset,
+        # the trainer state should be kept exactly the same as after the first fit
+        # to prove that fit_partial does not change trainer state before proceeding to training."
         assert t1 is not None
         assert t2 is not None
         assert_pl_trainers_equal(t1, t2)

Original file line number	Diff line number	Diff line change
`@@ -483,6 +483,9 @@ def on_train_epoch_start(self) -> None:`
`483`	`483`	`)`
`484`	`484`	`t2 = deepcopy(model.fit_trainer)`
`485`	`485`
	`486`	`+ # Since for the second we are fitting on an empty dataset,`
	`487`	`+ # the trainer state should be kept exactly the same as after the first fit`
	`488`	`+ # to prove that fit_partial does not change trainer state before proceeding to training."`
`486`	`489`	`assert t1 is not None`
`487`	`490`	`assert t2 is not None`
`488`	`491`	`assert_pl_trainers_equal(t1, t2)`