From 346b963137d1b31229acaf1bb89798669e652cad Mon Sep 17 00:00:00 2001
From: hong <weih1121@qq.com>
Date: Fri, 3 Jan 2025 11:40:17 +0800
Subject: [PATCH] wip

---
 sky/backends/backend_utils.py        |  1 +
 sky/backends/cloud_vm_ray_backend.py | 13 ++++++-------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/sky/backends/backend_utils.py b/sky/backends/backend_utils.py
index e70cf6200fd..e406cf6c435 100644
--- a/sky/backends/backend_utils.py
+++ b/sky/backends/backend_utils.py
@@ -807,6 +807,7 @@ def write_cluster_config(
             if (remote_identity_config ==
                     schemas.RemoteIdentityOptions.NO_UPLOAD.value):
                 excluded_clouds.add(cloud_obj)
+
     credentials = sky_check.get_cloud_credential_file_mounts(excluded_clouds)
     auth_config = {'ssh_private_key': auth.PRIVATE_SSH_KEY_PATH}
     region_name = resources_vars.get('region')
diff --git a/sky/backends/cloud_vm_ray_backend.py b/sky/backends/cloud_vm_ray_backend.py
index 1063090a5e3..3dff7c0e996 100644
--- a/sky/backends/cloud_vm_ray_backend.py
+++ b/sky/backends/cloud_vm_ray_backend.py
@@ -21,7 +21,6 @@
 from typing import (Any, Callable, Dict, Iterable, List, Optional, Set, Tuple,
                     Union)
 
-import click
 import colorama
 import filelock
 
@@ -2008,12 +2007,12 @@ def provision_with_retries(
                 enabled_clouds)
 
             if len(expirable_clouds) > 0:
-                warnings = (f'\nWarning: Expiring credentials detected for '
-                            f'{expirable_clouds}. Clusters may be leaked if '
-                            f'the credentials expire while jobs are running. '
-                            f'It is recommended to use credentials that never'
-                            f' expire or a service account.')
-                click.secho(warnings, fg='yellow')
+                warnings = (f'\n\033[93m Warning: Credentials used for '
+                            f'{expirable_clouds} may expire.Clusters may be '
+                            f'leaked if the credentials expire while jobs '
+                            f'are running.It is recommended to use credentials'
+                            f' that never expire or a service account.\033[0m')
+                logger.warning(warnings)
 
         # Retrying launchable resources.
         while True: