tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix test and pylint checks

Browse Source
This commit is contained in:
Wing Lian
2023-08-07 09:36:29 -04:00
parent 394a65f11f
commit 229b9165aa
3 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/axolotl/prompt_strategies/alpaca_w_system.py
View File

@@ -90,6 +90,7 @@ class OpenOrcaSystemDataPrompter(SystemDataPrompter):
""" """
def match_prompt_style(self): def match_prompt_style(self):
# pylint: disable=duplicate-code
if self.prompt_style == PromptStyle.INSTRUCT.value: if self.prompt_style == PromptStyle.INSTRUCT.value:
self.turn_format = "### User:\n{instruction}\n\n### Additional Context:\n{input}\n\n### Assistant:\n" self.turn_format = "### User:\n{instruction}\n\n### Additional Context:\n{input}\n\n### Assistant:\n"
self.turn_no_input_format = "### User:\n{instruction}\n\n### Assistant:\n" self.turn_no_input_format = "### User:\n{instruction}\n\n### Assistant:\n"

1
src/axolotl/prompters.py
View File

@@ -36,6 +36,7 @@ class AlpacaPrompter:
self.match_prompt_style() self.match_prompt_style()
def match_prompt_style(self): def match_prompt_style(self):
# pylint: disable=duplicate-code
if self.prompt_style == PromptStyle.INSTRUCT.value: if self.prompt_style == PromptStyle.INSTRUCT.value:
self.turn_format = "### Instruction:\n{instruction}\n\n### Input:\n{input}\n\n### Response:\n" self.turn_format = "### Instruction:\n{instruction}\n\n### Input:\n{input}\n\n### Response:\n"
self.turn_no_input_format = ( self.turn_no_input_format = (

2
tests/test_prompters.py
View File

@@ -70,7 +70,7 @@ class AlpacaPrompterTest(unittest.TestCase):
) )
) )
assert "use cot" in res assert "use cot" in res
assert res.startswith("### System:") assert res.startswith("SYSTEM:")
assert "### Instruction:" not in res assert "### Instruction:" not in res
assert "### Input:" not in res assert "### Input:" not in res
assert "alpacas" in res assert "alpacas" in res