Fix tests

huggingface · Apr 16, 2024 · 53f83a5 · 53f83a5
1 parent 4f1654c
commit 53f83a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/training_args.py b/src/transformers/training_args.py
@@ -734,7 +734,7 @@ class TrainingArguments:
     do_eval: bool = field(default=False, metadata={"help": "Whether to run eval on the dev set."})
     do_predict: bool = field(default=False, metadata={"help": "Whether to run predictions on the test set."})
     eval_strategy: Union[IntervalStrategy, str] = field(
-        default=None,
+        default="no",
         metadata={"help": "The evaluation strategy to use."},
     )
     prediction_loss_only: bool = field(
@@ -1269,7 +1269,7 @@ class TrainingArguments:
             "choices": ["auto", "apex", "cpu_amp"],
         },
     )
-    evaluation_strategy: Optional[str] = field(
+    evaluation_strategy: Union[IntervalStrategy, str] = field(
         default=None,
         metadata={"help": "Deprecated. Use `eval_strategy` instead"},
     )