From adf33d13934144b209b65f8e81b401475c407768 Mon Sep 17 00:00:00 2001 From: Sherin Thomas Date: Mon, 19 Dec 2022 07:20:22 +0000 Subject: [PATCH] fixes to imports --- pyproject.toml | 2 +- tests/tests_app/components/serve/test_auto_scaler.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 8611ef9323deb..4461d956634c6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -79,8 +79,8 @@ module = [ "lightning_app.components.serve.types.image", "lightning_app.components.serve.types.type", "lightning_app.components.serve.python_server", + "lightning_app.components.serve.auto_scaler", "lightning_app.components.training", - "lightning_app.components.auto_scaler", "lightning_app.core.api", "lightning_app.core.app", "lightning_app.core.flow", diff --git a/tests/tests_app/components/serve/test_auto_scaler.py b/tests/tests_app/components/serve/test_auto_scaler.py index d1adb1c5aacd2..6bd5aa958b6bf 100644 --- a/tests/tests_app/components/serve/test_auto_scaler.py +++ b/tests/tests_app/components/serve/test_auto_scaler.py @@ -33,8 +33,8 @@ def test_num_replicas_after_init(): @patch("uvicorn.run") -@patch("lightning_app.components.auto_scaler._LoadBalancer.url") -@patch("lightning_app.components.auto_scaler.AutoScaler.num_pending_requests") +@patch("lightning_app.components.serve.auto_scaler._LoadBalancer.url") +@patch("lightning_app.components.serve.auto_scaler.AutoScaler.num_pending_requests") def test_num_replicas_not_above_max_replicas(*_): """Test self.num_replicas doesn't exceed max_replicas.""" max_replicas = 6 @@ -53,8 +53,8 @@ def test_num_replicas_not_above_max_replicas(*_): @patch("uvicorn.run") -@patch("lightning_app.components.auto_scaler._LoadBalancer.url") -@patch("lightning_app.components.auto_scaler.AutoScaler.num_pending_requests") +@patch("lightning_app.components.serve.auto_scaler._LoadBalancer.url") +@patch("lightning_app.components.serve.auto_scaler.AutoScaler.num_pending_requests") def test_num_replicas_not_belo_min_replicas(*_): """Test self.num_replicas doesn't exceed max_replicas.""" min_replicas = 1