diff --git a/.gitignore b/.gitignore index 3969475..500b88f 100644 --- a/.gitignore +++ b/.gitignore @@ -15,3 +15,4 @@ RAG/tmp.py Carrot-Assistant/omop_tmp.py RAG/.cache/ *.qdrant +/Carrot-Assistant/tests/log diff --git a/Carrot-Assistant/tests/test_evals.py b/Carrot-Assistant/tests/test_evals.py index a823fd3..f2ab72b 100644 --- a/Carrot-Assistant/tests/test_evals.py +++ b/Carrot-Assistant/tests/test_evals.py @@ -85,7 +85,7 @@ def llm_prompt(self): @pytest.fixture def llm_pipeline(self, llm_prompt): - return LLMPipeline(LLMModel["llama-3.1-8b"], llm_prompt) + return LLMPipeline(LLMModel.LLAMA_3_1_8B, llm_prompt) def test_returns_string(self, llm_pipeline): model_output = llm_pipeline.run({"input_sentence": "Polly wants a cracker"})