Update Multinode Warning (#16091)

Lightning-AI · Dec 16, 2022 · 3d509f6 · 3d509f6
1 parent 4c91dec
commit 3d509f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/lightning_app/components/multi_node/base.py b/src/lightning_app/components/multi_node/base.py
@@ -56,12 +56,12 @@ def run(
         """
         super().__init__()
         if num_nodes > 1 and not is_running_in_cloud():
-            num_nodes = 1
             warnings.warn(
                 f"You set {type(self).__name__}(num_nodes={num_nodes}, ...)` but this app is running locally."
                 " We assume you are debugging and will ignore the `num_nodes` argument."
                 " To run on multiple nodes in the cloud, launch your app with `--cloud`."
             )
+            num_nodes = 1
         self.ws = structures.List(
             *[
                 work_cls(

diff --git a/tests/tests_app/components/multi_node/test_base.py b/tests/tests_app/components/multi_node/test_base.py
@@ -13,7 +13,7 @@ class Work(LightningWork):
         def run(self):
             pass
 
-    with pytest.warns(UserWarning, match=escape("You set MultiNode(num_nodes=1, ...)` but ")):
+    with pytest.warns(UserWarning, match=escape("You set MultiNode(num_nodes=2, ...)` but ")):
         MultiNode(Work, num_nodes=2, cloud_compute=CloudCompute("gpu"))
 
     with no_warning_call(UserWarning, match=escape("You set MultiNode(num_nodes=1, ...)` but ")):