Skip to content

Commit 505a851

Browse files
address gemini comment
1 parent e8e5977 commit 505a851

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

keras_remote/backend/gke_client.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -461,14 +461,6 @@ def _check_node_pool_exists_cached(selector_items) -> bool:
461461
"cloud.google.com/gke-tpu-topology", ""
462462
)
463463

464-
# Infer accelerator count from machine type using registry
465-
# This is robust because it uses the same source of truth as the Pod spec generation
466-
for tpu_spec in accelerators.TPUS.values():
467-
for chips, topo_spec in tpu_spec.topologies.items():
468-
if topo_spec.machine_type == machine_type:
469-
pool_labels["cloud.google.com/gke-accelerator-count"] = str(chips)
470-
break
471-
472464
if all(pool_labels.get(k) == str(v) for k, v in selector.items()):
473465
return True
474466
return False

0 commit comments

Comments
 (0)