Skip to content

Commit

Permalink
Patch report_to test after #570
Browse files Browse the repository at this point in the history
  • Loading branch information
tomaarsen committed Jan 13, 2025
1 parent d9bf753 commit b961717
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_training_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"])

Expand Down

0 comments on commit b961717

Please sign in to comment.