Fixes Issue #2781 for gcloud command in getting number of workers

MrGeislinger · MrGeislinger · commit d1a7c12acbfe · 2025-12-03T18:46:42.000Z
diff --git a/tools/orchestration/multihost_runner.py b/tools/orchestration/multihost_runner.py
@@ -172,6 +172,7 @@ def get_slices():
       "gcloud",
       "compute",
       "tpus",
+      "tpu-vm",
       "describe",
       slice_names[0],
       "--flatten=networkEndpoints[]",
