tests: base.test_reader.py

datasciencecampus · Aug 20, 2024 · 0d69f64 · 0d69f64
1 parent 0c7659f
commit 0d69f64
Show file tree

Hide file tree

Showing 2 changed files with 412 additions and 1 deletion.
diff --git a/tests/common.py b/tests/common.py
@@ -52,7 +52,7 @@ def default_llm() -> ChatOllama:
     """Instantiate default LLM object for use in testing."""
 
     llm = ChatOllama(
-        model_name="gemma",
+        model="gemma",
         temperature=0,
         # max_output_tokens=2048,
     )