[VirtualCluster] vcluster cli command support minimal install (#467)

Signed-off-by: sule <[email protected]>
antgroup · Jan 16, 2025 · fde6dcc · fde6dcc
1 parent 11697d4
commit fde6dcc
Show file tree

Hide file tree

Showing 3 changed files with 314 additions and 208 deletions.
diff --git a/python/ray/dashboard/modules/virtual_cluster/cli.py b/python/ray/dashboard/modules/virtual_cluster/cli.py
diff --git a/python/ray/dashboard/state_aggregator.py b/python/ray/dashboard/state_aggregator.py
@@ -512,7 +512,7 @@ def transform(reply) -> ListApiResponse:
             for id, entry in entries.items():
                 for node_id in cluster_nodes[id]:
                     node = all_nodes[node_id]
-                    if not node["is_dead"]:
+                    if not node["is_dead"] and node["template_id"] != "":
                         entry["replica_sets"][node["template_id"]] = (
                             entry["replica_sets"].get(node["template_id"], 0) + 1
                         )
@@ -533,7 +533,7 @@ def collect_all_sub_nodes(virtual_cluster_id):
                 for node_id, node in entry["visible_node_instances"].items():
                     if node_id not in divided_nodes:
                         undivided_nodes[node_id] = node
-                        if not node["is_dead"]:
+                        if not node["is_dead"] and node["template_id"] != "":
                             entry["undivided_replica_sets"][node["template_id"]] = (
                                 entry["undivided_replica_sets"].get(
                                     node["template_id"], 0