Add ipython role to learn

truefoundry · Aug 20, 2024 · 68a0c90 · 68a0c90
1 parent 31093ec
commit 68a0c90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_utils.py b/data_utils.py
@@ -63,7 +63,7 @@ def _make_dataset_file_source(
             "message_field_content": "content",
             "roles": {"system": ["system"], "user": ["user", "human"], "assistant": ["assistant"], "tool": ["tool"]},
             "split": split,
-            "roles_to_train": ["gpt", "assistant"],
+            "roles_to_train": ["gpt", "assistant", "ipython"],
             "train_on_eos": "last",
         }
     else: