Fix incorrect node count when tpu is in the cluster name (#46)

AI-Hypercomputer · Dec 8, 2023 · 67449dd · 67449dd
1 parent 3afdaf8
commit 67449dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xpk.py b/xpk.py
@@ -1591,7 +1591,7 @@ def cluster_describe(args) -> int:
     xpk_exit(return_code)
 
   return_code_node_output, node_output = run_command_for_value(
-      r"kubectl get node --no-headers=true | grep '\-tpu\-' | wc -l",
+      r"kubectl get node --no-headers=true --selector='cloud.google.com/gke-tpu-accelerator' | wc -l",
       'Count TPU Nodes',
       args,
   )