fix: write and delete models for testing in a temporary directory

umbertogriffo · Jan 11, 2025 · 7204cad · 7204cad
1 parent 3345129
commit 7204cad
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/bot/client/test_lamacpp_client.py b/tests/bot/client/test_lamacpp_client.py
@@ -23,9 +23,9 @@ def model_settings():
 
 
 @pytest.fixture
-def lamacpp_client(mock_model_folder, model_settings, cpu_config):
+def lamacpp_client(mock_models_folder, model_settings, cpu_config):
     with patch.object(model_settings, "config", cpu_config):
-        return LamaCppClient(mock_model_folder, model_settings)
+        return LamaCppClient(mock_models_folder, model_settings)
 
 
 def test_generate_answer(lamacpp_client):

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -4,7 +4,7 @@
 
 
 @pytest.fixture
-def mock_model_folder(tmp_path):
-    model_folder = tmp_path / "models"
-    Path(model_folder).mkdir(parents=True, exist_ok=True)
-    return model_folder
+def mock_models_folder(tmp_path):
+    models_folder = tmp_path / "models"
+    Path(models_folder).mkdir()
+    return models_folder