diff --git a/tests/test_training_args.py b/tests/test_training_args.py index ecce4f42..5e035cd7 100644 --- a/tests/test_training_args.py +++ b/tests/test_training_args.py @@ -64,9 +64,9 @@ def test_learning_rates(self): def test_report_to(self): args = TrainingArguments(report_to="none") - self.assertEqual(args.report_to, []) + self.assertEqual(args.report_to, ["none"]) args = TrainingArguments(report_to=["none"]) - self.assertEqual(args.report_to, []) + self.assertEqual(args.report_to, ["none"]) args = TrainingArguments(report_to="hello") self.assertEqual(args.report_to, ["hello"])