From 97b09c37f1fd233a53335a7ec1d2880788098086 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: 393cc977573c21171b6ef94e1045ef99488760b1 --- 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")