Fix coverage [release]

datadreamer-dev · Mar 18, 2024 · ad3dd9c · ad3dd9c
1 parent 8f3b065
commit ad3dd9c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "DataDreamer"
-version = "0.24.0"
+version = "0.25.0"
 description = "Prompt. Generate Synthetic Data. Train & Align Models."
 license = "MIT"
 authors= [

diff --git a/src/tests/trainers/test_trainers.py b/src/tests/trainers/test_trainers.py
@@ -169,7 +169,9 @@ def test_hf_chat_template(self, create_datadreamer):
             )
         with create_datadreamer():
             trainer = TrainHFFineTune(
-                "Zephyr Trainer", model_name="HuggingFaceH4/zephyr-7b-beta"
+                "Zephyr Trainer",
+                model_name="HuggingFaceH4/zephyr-7b-beta",
+                chat_prompt_template="<|system|>\n</s>\n<|user|>\n{{prompt}}</s>\n<|assistant|>\n",
             )
             assert (
                 trainer.tokenizer.apply_chat_template(
@@ -179,7 +181,7 @@ def test_hf_chat_template(self, create_datadreamer):
                     ],
                     tokenize=False,
                 )
-                == "<s><|system|>\nYou are a helpful assistant.</s>\n<|user|>\nA</s>\n<|assistant|>\nB</s>"
+                == "<s><|system|>\n</s>\n<|user|>\nA</s>\n<|assistant|>\nB</s>"
             )
         with create_datadreamer():
             trainer = TrainHFFineTune(