Fix TPU VM pod stop

skypilot-org · Dec 30, 2023 · 925a35a · 925a35a
1 parent 708d069
commit 925a35a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sky/clouds/gcp.py b/sky/clouds/gcp.py
@@ -167,7 +167,7 @@ class GCP(clouds.Cloud):
     def _unsupported_features_for_resources(
         cls, resources: 'resources.Resources'
     ) -> Dict[clouds.CloudImplementationFeatures, str]:
-        if tpu_utils.is_tpu_vm_pod(resources):
+        if gcp_utils.is_tpu_vm_pod(resources):
             return {
                 clouds.CloudImplementationFeatures.STOP: (
                     'TPU VM pods cannot be stopped. Please refer to: https://cloud.google.com/tpu/docs/managing-tpus-tpu-vm#stopping_your_resources'