Fix failing tests

huggingface · Apr 24, 2024 · e46a42e · e46a42e
1 parent 865bf2a
commit e46a42e
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/tests/trainer/test_trainer_callback.py b/tests/trainer/test_trainer_callback.py
@@ -269,7 +269,7 @@ def test_stateful_callbacks(self):
             callbacks=[cb],
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -281,7 +281,7 @@ def test_stateful_callbacks(self):
             callbacks=[EarlyStoppingCallback()],
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -307,7 +307,7 @@ def test_stateful_mixed_callbacks(self):
             callbacks=cbs,
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -319,7 +319,7 @@ def test_stateful_mixed_callbacks(self):
             callbacks=[EarlyStoppingCallback(), MyTestTrainerCallback()],
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -346,7 +346,7 @@ def test_stateful_duplicate_callbacks(self):
             callbacks=cbs,
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -358,7 +358,7 @@ def test_stateful_duplicate_callbacks(self):
             callbacks=[MyTestExportableCallback(), MyTestExportableCallback()],
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -382,7 +382,7 @@ def test_missing_stateful_callback(self):
             callbacks=[cb],
             load_best_model_at_end=True,
             save_strategy="steps",
-            evaluation_strategy="steps",
+            eval_strategy="steps",
             save_steps=2,
             eval_steps=2,
             max_steps=2,
@@ -391,6 +391,10 @@ def test_missing_stateful_callback(self):
 
         # Create a new trainer with defaults
         trainer = self.get_trainer(
+            save_strategy="steps",
+            eval_strategy="steps",
+            save_steps=2,
+            eval_steps=2,
             max_steps=2,
             restore_callback_states_from_checkpoint=True,
         )