[CI] fix dpo gpu ci tests (#2189)

* fix dpo ci test * color-blind
huggingface · Oct 7, 2024 · 51ca76b · 51ca76b
1 parent 7005ab4
commit 51ca76b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/scripts/log_example_reports.py b/scripts/log_example_reports.py
@@ -55,7 +55,7 @@ def main(text_file_name, slack_channel_name=None):
             "type": "section",
             "text": {
                 "type": "plain_text",
-                "text": "🔴 Something is wrong with the workflow please check ASAP!"
+                "text": "❌ Something is wrong with the workflow please check ASAP!"
                 "Something went wrong there is no text file being produced. Please check ASAP.",
                 "emoji": True,
             },
@@ -82,7 +82,7 @@ def main(text_file_name, slack_channel_name=None):
 
         for test_name, failed in final_results.items():
             failed_table = tabulate(
-                [[test_name, "🟢" if not failed else "🔴"]],
+                [[test_name, "✅" if not failed else "❌"]],
                 headers=["Test Name", "Status"],
                 showindex="always",
                 tablefmt="grid",

diff --git a/tests/slow/test_dpo_slow.py b/tests/slow/test_dpo_slow.py
@@ -85,9 +85,9 @@ def test_dpo_bare_model(self, model_id, loss_type, pre_compute_logits):
                 model=model,
                 ref_model=None,
                 args=training_args,
+                train_dataset=self.dataset["train"],
+                eval_dataset=self.dataset["test"],
                 processing_class=tokenizer,
-                train_dataset=self.dataset,
-                eval_dataset=self.dataset,
             )
 
             # train the model
@@ -142,9 +142,9 @@ def test_dpo_peft_model(self, model_id, loss_type, pre_compute_logits, gradient_
                 model=model,
                 ref_model=None,
                 args=training_args,
+                train_dataset=self.dataset["train"],
+                eval_dataset=self.dataset["test"],
                 processing_class=tokenizer,
-                train_dataset=self.dataset,
-                eval_dataset=self.dataset,
                 peft_config=self.peft_config,
             )
 
@@ -206,9 +206,9 @@ def test_dpo_peft_model_qlora(self, model_id, loss_type, pre_compute_logits, gra
                 model=model,
                 ref_model=None,
                 args=training_args,
+                train_dataset=self.dataset["train"],
+                eval_dataset=self.dataset["test"],
                 processing_class=tokenizer,
-                train_dataset=self.dataset,
-                eval_dataset=self.dataset,
                 peft_config=self.peft_config,
             )