🧪 fix test

shroominic · Nov 15, 2023 · bc608b0 · bc608b0
1 parent 2e3b7f4
commit bc608b0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/extraction_test.py b/tests/extraction_test.py
@@ -1,4 +1,6 @@
-from funcchain import BaseModel, achain, chain
+from funcchain import achain, chain, settings, BaseModel
+
+settings.MODEL_TEMPERATURE = 0
 
 
 class Task(BaseModel):
@@ -47,9 +49,9 @@ def test_extraction() -> None:
     )
 
     task_description = description(task)
-    task2 = _await(extract_task(task_description))
+    _task = _await(extract_task(task_description))
 
-    assert compare_tasks(task, task2)
+    assert compare_tasks(task, _task)
 
 
 if __name__ == "__main__":