From 3a496ce7aa146da05f669225b8c7995f92b29788 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 15 Nov 2023 17:20:24 -0500 Subject: [PATCH] module path error autoscaler fixes Former-commit-id: 08112384076420c4bc9d87194f81e4835ad5d55f --- tests/swarms/autoscaler.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/swarms/autoscaler.py b/tests/swarms/autoscaler.py index bd884d73..976b5b23 100644 --- a/tests/swarms/autoscaler.py +++ b/tests/swarms/autoscaler.py @@ -45,8 +45,8 @@ def test_autoscaler_scale_down(): assert len(autoscaler.agents_pool) == 4 -@patch("your_module.AutoScaler.scale_up") -@patch("your_module.AutoScaler.scale_down") +@patch("swarms.swarms.AutoScaler.scale_up") +@patch("swarms.swarms.AutoScaler.scale_down") def test_autoscaler_monitor_and_scale(mock_scale_down, mock_scale_up): autoscaler = AutoScaler(initial_agents=5, agent=flow) autoscaler.add_task("task1") @@ -55,8 +55,8 @@ def test_autoscaler_monitor_and_scale(mock_scale_down, mock_scale_up): mock_scale_down.assert_called_once() -@patch("your_module.AutoScaler.monitor_and_scale") -@patch("your_module.flow.run") +@patch("swarms.swarms.AutoScaler.monitor_and_scale") +@patch("swarms.swarms.flow.run") def test_autoscaler_start(mock_run, mock_monitor_and_scale): autoscaler = AutoScaler(initial_agents=5, agent=flow) autoscaler.add_task("task1")