diff --git a/sky/cli.py b/sky/cli.py index 1fcb9786f17..191e56cd910 100644 --- a/sky/cli.py +++ b/sky/cli.py @@ -1063,10 +1063,7 @@ def _make_task_or_dag_from_entrypoint_with_overrides( # Spot launch specific. if spot_recovery is not None: - if spot_recovery.lower() == 'none': - override_params['spot_recovery'] = None - else: - override_params['spot_recovery'] = spot_recovery + override_params['spot_recovery'] = spot_recovery assert len(task.resources) == 1 old_resources = list(task.resources)[0] diff --git a/sky/resources.py b/sky/resources.py index c36fee9d9b7..e50bac16c90 100644 --- a/sky/resources.py +++ b/sky/resources.py @@ -128,7 +128,8 @@ def __init__( self._use_spot = use_spot if use_spot is not None else False self._spot_recovery = None if spot_recovery is not None: - self._spot_recovery = spot_recovery.upper() + if spot_recovery.strip().lower() != 'none': + self._spot_recovery = spot_recovery.upper() if disk_size is not None: if round(disk_size) != disk_size: