Move dashboard parameters to initial Ray startup command.

mobiusml · Nov 22, 2024 · 0473053 · 0473053
1 parent ed16a5d
commit 0473053
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aana/sdk.py b/aana/sdk.py
@@ -106,9 +106,6 @@ def connect(
                 address=address,
                 ignore_reinit_error=True,
                 log_to_driver=show_logs,
-                include_dashboard=True,
-                dashboard_host=dashboard_host,
-                dashboard_port=dashboard_port,
             )
         except ConnectionError:
             # If connection fails, start a new Ray cluster and serve instance
@@ -117,6 +114,9 @@ def connect(
                 log_to_driver=show_logs,
                 num_cpus=num_cpus,
                 num_gpus=num_gpus,
+                include_dashboard=True,
+                dashboard_host=dashboard_host,
+                dashboard_port=dashboard_port,
             )
 
         serve_status = serve.status()