CI: AMD MI300 tests fix (#30797)

* add fix * update import * updated dicts and comments * remove prints * Update testing_utils.py
2024-05-21 17:16:07 +05:30
parent a755745546
commit 7a4792e6b3
10 changed files with 126 additions and 10 deletions
--- a/tests/trainer/test_trainer_seq2seq.py
+++ b/tests/trainer/test_trainer_seq2seq.py
@@ -161,7 +161,6 @@ class Seq2seqTrainerTester(TestCasePlus):
            tokenizer=tokenizer,
            data_collator=data_collator,
            compute_metrics=lambda x: {"samples": x[0].shape[0]},
-            report_to="none",
        )

        def prepare_data(examples):