From b9617171d281c2101487cf6b0ab81c14807b7110 Mon Sep 17 00:00:00 2001 From: Tom Aarsen Date: Mon, 13 Jan 2025 11:36:36 +0100 Subject: [PATCH] Patch report_to test after #570 --- tests/test_training_args.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"])