force MultiAgentsTests to dump data into tmp dir, not into repo

huggingface · Feb 13, 2025 · 0b82035 · 0b82035
1 parent 8b247fe
commit 0b82035
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_agents.py b/tests/test_agents.py
@@ -32,7 +32,7 @@
     populate_template,
 )
 from smolagents.default_tools import DuckDuckGoSearchTool, PythonInterpreterTool, VisitWebpageTool
-from smolagents.memory import PlanningStep, ActionStep
+from smolagents.memory import ActionStep, PlanningStep
 from smolagents.models import (
     ChatMessage,
     ChatMessageToolCall,
@@ -796,6 +796,10 @@ def test_call_with_provide_run_summary(self, provide_run_summary):
 
 
 class MultiAgentsTests(unittest.TestCase):
+    @pytest.fixture(autouse=True)
+    def initdir(self, tmp_path, monkeypatch):
+        monkeypatch.chdir(tmp_path)
+
     def test_multiagents_save(self):
         model = HfApiModel("Qwen/Qwen2.5-Coder-32B-Instruct", max_tokens=2096, temperature=0.5)