wrap rename

sky/provision/do/instance.py

@@ -86,13 +86,9 @@ def run_instances(region: str, cluster_name_on_cloud: str,
     if to_start_count == 0:
         if head_instance is None:
             head_instance = list(exist_instances.values())[0]
-            utils.client().droplet_actions.rename(
-                droplet=head_instance['id'],
-                body={
-                    'type': 'rename',
-                    'name': f'{cluster_name_on_cloud}-'
-                            f'{uuid.uuid4().hex[:4]}-head'
-                })
+            utils.rename_instance(
+                head_instance,
+                f'{cluster_name_on_cloud}-{uuid.uuid4().hex[:4]}-head')
         assert head_instance is not None, ('`head_instance` should not be None')
         logger.info(f'Cluster {cluster_name_on_cloud} already has '
                     f'{len(exist_instances)} nodes, no need to start more.')

sky/provision/do/utils.py

@@ -54,8 +54,7 @@ def _init_client():
             'no credentials file found from '
             f'the following paths {POSSIBLE_CREDENTIALS_PATHS}')
 
-    auth_contexts = common_utils.read_yaml(
-        CREDENTIALS_PATH)['auth-contexts']
+    auth_contexts = common_utils.read_yaml(CREDENTIALS_PATH)['auth-contexts']
     for context, api_token in auth_contexts.items():
         try:
             test_client = do.pydo.Client(token=api_token)
@@ -204,6 +203,14 @@ def down_instance(instance: Dict[str, Any]):
         droplet_id=instance['id'], x_dangerous=True)
 
 
+def rename_instance(instance: Dict[str, Any], new_name: str):
+    client().droplet_actions.rename(droplet=instance['id'],
+                                    body={
+                                        'type': 'rename',
+                                        'name': new_name
+                                    })
+
+
 def filter_instances(
         cluster_name_on_cloud: str,
         status_filters: Optional[List[str]] = None) -> Dict[str, Any]: