Allow autodown for k8s

skypilot-org · May 8, 2024 · 52e8aba · 52e8aba
1 parent 12c156a
commit 52e8aba
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/sky/backends/cloud_vm_ray_backend.py b/sky/backends/cloud_vm_ray_backend.py
@@ -1994,10 +1994,9 @@ def provision_with_retries(
 
  requested_features = self._requested_features.copy()
  # Skip stop feature for Kubernetes jobs controller.
- if isinstance(to_provision.cloud, clouds.Kubernetes
- ) and controller_utils.Controllers.from_name(
- cluster_name
- ) == controller_utils.Controllers.JOBS_CONTROLLER:
+ if (isinstance(to_provision.cloud, clouds.Kubernetes) and
+ controller_utils.Controllers.from_name(cluster_name)
+ == controller_utils.Controllers.JOBS_CONTROLLER):
  assert (clouds.CloudImplementationFeatures.STOP
  in requested_features), requested_features
  requested_features.remove(
@@ -4066,7 +4065,8 @@ def set_autostop(self,
  # cloud and resources support requested autostop.
  if idle_minutes_to_autostop is not None:
  # Skip auto-stop for Kubernetes clusters.
- if isinstance(handle.launched_resources.cloud, clouds.Kubernetes):
+ if (isinstance(handle.launched_resources.cloud, clouds.Kubernetes)
+ and not down):
  # We should hit this code path only for the jobs controller on
  # Kubernetes clusters.
  assert (controller_utils.Controllers.from_name(