add split to dpo tests

axolotl-ai-cloud · Jan 23, 2024 · 7f3b7ce · 7f3b7ce
1 parent 44a6f2d
commit 7f3b7ce
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/e2e/test_dpo.py b/tests/e2e/test_dpo.py
@@ -44,6 +44,7 @@ def test_dpo_lora(self, temp_dir):
                     {
                         "path": "Intel/orca_dpo_pairs",
                         "type": "chatml.intel",
+                        "split": "train",
                     },
                 ],
                 "num_epochs": 1,
@@ -84,6 +85,7 @@ def test_kto_pair_lora(self, temp_dir):
                     {
                         "path": "Intel/orca_dpo_pairs",
                         "type": "chatml.intel",
+                        "split": "train",
                     },
                 ],
                 "num_epochs": 1,
@@ -124,6 +126,7 @@ def test_ipo_lora(self, temp_dir):
                     {
                         "path": "Intel/orca_dpo_pairs",
                         "type": "chatml.intel",
+                        "split": "train",
                     },
                 ],
                 "num_epochs": 1,