From 2637c4dbea66918a8bba74aee49b6cefa1574e6f Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Sun, 6 Oct 2024 21:04:25 +0200 Subject: [PATCH 1/2] fix dpo ci test --- tests/slow/test_dpo_slow.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/slow/test_dpo_slow.py b/tests/slow/test_dpo_slow.py index 96e7af086d..37e0aa754a 100644 --- a/tests/slow/test_dpo_slow.py +++ b/tests/slow/test_dpo_slow.py @@ -86,8 +86,8 @@ def test_dpo_bare_model(self, model_id, loss_type, pre_compute_logits): ref_model=None, args=training_args, tokenizer=tokenizer, - train_dataset=self.dataset, - eval_dataset=self.dataset, + train_dataset=self.dataset["train"], + eval_dataset=self.dataset["test"], ) # train the model @@ -143,8 +143,8 @@ def test_dpo_peft_model(self, model_id, loss_type, pre_compute_logits, gradient_ ref_model=None, args=training_args, tokenizer=tokenizer, - train_dataset=self.dataset, - eval_dataset=self.dataset, + train_dataset=self.dataset["train"], + eval_dataset=self.dataset["test"], peft_config=self.peft_config, ) @@ -207,8 +207,8 @@ def test_dpo_peft_model_qlora(self, model_id, loss_type, pre_compute_logits, gra ref_model=None, args=training_args, tokenizer=tokenizer, - train_dataset=self.dataset, - eval_dataset=self.dataset, + train_dataset=self.dataset["train"], + eval_dataset=self.dataset["test"], peft_config=self.peft_config, ) From 5d085ae2744f420fc42ae88feaec44a57921894f Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Sun, 6 Oct 2024 22:03:17 +0200 Subject: [PATCH 2/2] color-blind --- scripts/log_example_reports.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/log_example_reports.py b/scripts/log_example_reports.py index b49a608826..fa31fbf589 100644 --- 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",