format

skypilot-org · Nov 13, 2022 · 5d1af1a · 5d1af1a
1 parent e53c444
commit 5d1af1a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/sky/spot/controller.py b/sky/spot/controller.py
@@ -92,8 +92,8 @@ def _run(self):
                     # job may not FAILED immediately when part of the nodes are
                     # preempted.
                     (cluster_status,
-                        handle) = backend_utils.refresh_cluster_status_handle(
-                            self._cluster_name, force_refresh=True)
+                     handle) = backend_utils.refresh_cluster_status_handle(
+                         self._cluster_name, force_refresh=True)
                     if cluster_status != global_user_state.ClusterStatus.UP:
                         # recover the cluster if it is not up.
                         need_recovery = True

diff --git a/sky/spot/spot_utils.py b/sky/spot/spot_utils.py
@@ -265,7 +265,8 @@ def stream_logs_by_id(job_id: int, follow: bool = True) -> str:
                             f'(status: {job_status.value}).')
                 break
             logger.info('INFO: (Log streaming) The job is cancelled. Waiting '
-                        'for the controller process to update the status.')
+                        'for the controller process to update the status in '
+                        f'{JOB_STATUS_CHECK_GAP_SECONDS} seconds.')
         else:
             logger.info(
                 f'INFO: (Log streaming) Got return code {returncode}. Retrying '