bugfixes

shouldn't keep test logs forgot how to use enums
Health-Informatics-UoN · Oct 18, 2024 · 127dba3 · 127dba3
1 parent 9f76c55
commit 127dba3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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"})