test

mosaicml · Nov 29, 2024 · 8d1bd92 · 8d1bd92
1 parent dd39ec3
commit 8d1bd92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/algorithms/test_algorithms_train.py b/tests/algorithms/test_algorithms_train.py
@@ -18,7 +18,7 @@ def test_algorithm_trains(alg_cls: type[Algorithm]):
     trainer = Trainer(
         model=model,
         train_dataloader=dataloader,
-        max_duration='1ba',
+        max_duration='2ba',
         algorithms=alg_cls(**alg_kwargs),
     )
     trainer.fit()
@@ -34,5 +34,5 @@ def test_algorithm_trains(alg_cls: type[Algorithm]):
             'GyroDropout is implemented to be applied on Event.FIT_START, so is not compatible with multiple calls to fit.',
         )
 
-    # fit again for another epoch
+    # fit again for another batch
     trainer.fit(duration='1ba')