fix test and pylint checks

2023-08-07 09:36:29 -04:00
parent 394a65f11f
commit 229b9165aa
3 changed files with 3 additions and 1 deletions
--- a/tests/test_prompters.py
+++ b/tests/test_prompters.py
@@ -70,7 +70,7 @@ class AlpacaPrompterTest(unittest.TestCase):
            )
        )
        assert "use cot" in res
-        assert res.startswith("### System:")
+        assert res.startswith("SYSTEM:")
        assert "### Instruction:" not in res
        assert "### Input:" not in res
        assert "alpacas" in res