skypilot-nightly 1.0.0.dev20250224__py3-none-any.whl → 1.0.0.dev20250225__py3-none-any.whl

sky/__init__.py

@@ -5,7 +5,7 @@ from typing import Optional
 import urllib.request
 
 # Replaced with the current commit when building the wheels.
-_SKYPILOT_COMMIT_SHA = '73e95ab79bf75923963a93733e54c100d0567e02'
+_SKYPILOT_COMMIT_SHA = '546c0864e0f1e32e3d1080d9b7a5fdf293bc1ad9'
 
 
 def _get_git_commit():
@@ -35,7 +35,7 @@ def _get_git_commit():
 
 
 __commit__ = _get_git_commit()
-__version__ = '1.0.0.dev20250224'
+__version__ = '1.0.0.dev20250225'
 __root_dir__ = os.path.dirname(os.path.abspath(__file__))
 
 

sky/adaptors/nebius.py

@@ -5,8 +5,10 @@ from sky.adaptors import common
 
 NEBIUS_TENANT_ID_FILENAME = 'NEBIUS_TENANT_ID.txt'
 NEBIUS_IAM_TOKEN_FILENAME = 'NEBIUS_IAM_TOKEN.txt'
+NEBIUS_PROJECT_ID_FILENAME = 'NEBIUS_PROJECT_ID.txt'
 NEBIUS_TENANT_ID_PATH = '~/.nebius/' + NEBIUS_TENANT_ID_FILENAME
 NEBIUS_IAM_TOKEN_PATH = '~/.nebius/' + NEBIUS_IAM_TOKEN_FILENAME
+NEBIUS_PROJECT_ID_PATH = '~/.nebius/' + NEBIUS_PROJECT_ID_FILENAME
 
 MAX_RETRIES_TO_DISK_CREATE = 120
 MAX_RETRIES_TO_INSTANCE_STOP = 120
@@ -20,6 +22,7 @@ POLL_INTERVAL = 5
 
 _iam_token = None
 _tenant_id = None
+_project_id = None
 
 nebius = common.LazyImport(
     'nebius',
@@ -69,6 +72,18 @@ def get_iam_token():
     return _iam_token
 
 
+def get_project_id():
+    global _project_id
+    if _project_id is None:
+        try:
+            with open(os.path.expanduser(NEBIUS_PROJECT_ID_PATH),
+                      encoding='utf-8') as file:
+                _project_id = file.read().strip()
+        except FileNotFoundError:
+            return None
+    return _project_id
+
+
 def get_tenant_id():
     global _tenant_id
     if _tenant_id is None:

sky/clouds/nebius.py

@@ -15,7 +15,8 @@ if typing.TYPE_CHECKING:
 _CREDENTIAL_FILES = [
     # credential files for Nebius
     nebius.NEBIUS_TENANT_ID_FILENAME,
-    nebius.NEBIUS_IAM_TOKEN_FILENAME
+    nebius.NEBIUS_IAM_TOKEN_FILENAME,
+    nebius.NEBIUS_PROJECT_ID_FILENAME,
 ]
 
 
@@ -255,6 +256,8 @@ class Nebius(clouds.Cloud):
         token_msg = ('    Credentials can be set up by running: \n'\
                     f'        $ nebius iam get-access-token > {nebius.NEBIUS_IAM_TOKEN_PATH} \n')  # pylint: disable=line-too-long
         tenant_msg = ('   Copy your tenat ID from the web console and save it to file \n'  # pylint: disable=line-too-long
+                      f'        $ echo $NEBIUS_TENANT_ID_PATH > {nebius.NEBIUS_TENANT_ID_PATH} \n'  # pylint: disable=line-too-long
+                      '   Or if you have 1 tenant you can run:\n'  # pylint: disable=line-too-long
                       f'        $ nebius --format json iam whoami|jq -r \'.user_profile.tenants[0].tenant_id\' > {nebius.NEBIUS_TENANT_ID_PATH} \n')  # pylint: disable=line-too-long
         if token is None:
             return False, f'{token_msg}'

sky/global_user_state.py

@@ -60,7 +60,7 @@ def create_table(cursor, conn):
         last_use TEXT,
         status TEXT,
         autostop INTEGER DEFAULT -1,
-        metadata TEXT DEFAULT "{}",
+        metadata TEXT DEFAULT '{}',
         to_down INTEGER DEFAULT 0,
         owner TEXT DEFAULT null,
         cluster_hash TEXT DEFAULT null,
@@ -118,7 +118,7 @@ def create_table(cursor, conn):
                                  'INTEGER DEFAULT -1')
 
     db_utils.add_column_to_table(cursor, conn, 'clusters', 'metadata',
-                                 'TEXT DEFAULT "{}"')
+                                 'TEXT DEFAULT \'{}\'')
 
     db_utils.add_column_to_table(cursor, conn, 'clusters', 'to_down',
                                  'INTEGER DEFAULT 0')
@@ -271,7 +271,7 @@ def add_or_update_cluster(cluster_name: str,
         # Keep the old metadata value if it exists, otherwise set it to
         # default {}.
         'COALESCE('
-        '(SELECT metadata FROM clusters WHERE name=?), "{}"),'
+        '(SELECT metadata FROM clusters WHERE name=?), \'{}\'),'
         # Keep the old owner value if it exists, otherwise set it to
         # default null.
         'COALESCE('

sky/jobs/state.py

@@ -212,6 +212,7 @@ class ManagedJobStatus(enum.Enum):
     is dedicated to a managed job, i.e. there should always be enough resource
     to run the job and the job will be immediately transitioned to RUNNING.
 
+    You can see a state diagram for ManagedJobStatus in sky/jobs/README.md.
     """
     # PENDING: Waiting for the jobs controller to have a slot to run the
     # controller process.
@@ -332,6 +333,8 @@ class ManagedJobScheduleState(enum.Enum):
       to transition directly from WAITING or even INACTIVE to DONE if the job is
       cancelled.
 
+    You can see a state diagram in sky/jobs/README.md.
+
     There is no well-defined mapping from the managed job status to schedule
     state or vice versa. (In fact, schedule state is defined on the job and
     status on the task.)

sky/provision/nebius/utils.py

@@ -44,13 +44,21 @@ def get_project_by_region(region: str) -> str:
     # Project id looks like project-e00xxxxxxxxxxxxxx where
     # e00 - id of region 'eu-north1'
     # e01 - id of region 'eu-west1'
+    region_ids = {'eu-north1': 'e00', 'eu-west1': 'e01'}
     # TODO(SalikovAlex): fix when info about region will be in projects list
     # Currently, Nebius cloud supports 2 regions. We manually enumerate
     # them here. Reference: https://docs.nebius.com/overview/regions
+
+    #  Check is there project if in config
+    preferable_project_id = nebius.get_project_id()
+    if preferable_project_id is not None:
+        if preferable_project_id[8:11] == region_ids[region]:
+            return preferable_project_id
+        logger.warning(
+            f'Can\'t use customized NEBIUS_PROJECT_ID ({preferable_project_id})'
+            f' for region {region}. Please check if the project ID is correct.')
     for project in projects.items:
-        if region == 'eu-north1' and project.metadata.id[8:11] == 'e00':
-            return project.metadata.id
-        if region == 'eu-west1' and project.metadata.id[8:11] == 'e01':
+        if project.metadata.id[8:11] == region_ids[region]:
             return project.metadata.id
     raise Exception(f'No project found for region "{region}".')
 

{skypilot_nightly-1.0.0.dev20250224.dist-info → skypilot_nightly-1.0.0.dev20250225.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: skypilot-nightly
-Version: 1.0.0.dev20250224
+Version: 1.0.0.dev20250225
 Summary: SkyPilot: An intercloud broker for the clouds
 Author: SkyPilot Team
 License: Apache 2.0

{skypilot_nightly-1.0.0.dev20250224.dist-info → skypilot_nightly-1.0.0.dev20250225.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sky/__init__.py,sha256=8zed0-GYhvWXZwZFaUANPausBHOuBBm3jZpWo9MJsmE,6428
+sky/__init__.py,sha256=KSl1VCjCC3uU-BihR5aIRst73gWU1IOO1xhIx-4pmso,6428
 sky/admin_policy.py,sha256=hPo02f_A32gCqhUueF0QYy1fMSSKqRwYEg_9FxScN_s,3248
 sky/authentication.py,sha256=hCEqi77nprQEg3ktfRL51xiiw16zwZOmFEDB_Z7fWVU,22384
 sky/check.py,sha256=NDKx_Zm7YRxPjMv82wz3ESLnGIPljaACyqVdVNM0PzY,11258
@@ -8,7 +8,7 @@ sky/core.py,sha256=X83hdpPTiWyEJLamrguCd03PUjkRiGgqTFfEBEQkzWc,45471
 sky/dag.py,sha256=Yl7Ry26Vql5cv4YMz8g9kOUgtoCihJnw7c8NgZYakMY,3242
 sky/exceptions.py,sha256=cGepNlBkjjgvi3fazc3DbdYLKhhF_sHCuGX0-hu_QMQ,13685
 sky/execution.py,sha256=0M4RTEzWn-B9oz221XdZOIGH12XOACmNq0j-WGUT_No,28023
-sky/global_user_state.py,sha256=PlAhEkBQ9MNAaZs96R8BouhWk9iVghm6CNKSxjsyAAY,33242
+sky/global_user_state.py,sha256=sUDdSsJeiJkbgmZNwy8YGFK0XeNh-RBr1VDUvbmjf0g,33246
 sky/models.py,sha256=4xSW05BdDPEjW8Ubvj3VlVOVnzv0TbrolsFvR5R5v1U,638
 sky/optimizer.py,sha256=C82l9N3umdrJ2AaM-pSg0aK5rpOAX3lEAfFU7r6hqPo,60183
 sky/resources.py,sha256=f2Qo_Wt0kFruKmYm6cgYbICH_wn0Zkb8uIv6LA82SRs,72153
@@ -26,7 +26,7 @@ sky/adaptors/docker.py,sha256=_kzpZ0fkWHqqQAVVl0llTsCE31KYz3Sjn8psTBQHVkA,468
 sky/adaptors/gcp.py,sha256=OQ9RaqjR0r0iaWYpjvEtIx5vnEhyB4LhUCwbtdxsmVk,3115
 sky/adaptors/ibm.py,sha256=H87vD6izq_wQI8oQC7cx9iVtRgPi_QkAcrfa1Z3PNqU,4906
 sky/adaptors/kubernetes.py,sha256=fvNeqqyW4WbUDT-isU_fWLN-zqywayyRkVsj7sJ4Vc8,6645
-sky/adaptors/nebius.py,sha256=bZR7ybfqIyUbNFiMfwC7qpOmSWDM9uExxWhm-MT_PqU,2259
+sky/adaptors/nebius.py,sha256=JOvwniQT-Pkp9-af6IdL_FUkjIbsEAUXVNUkwdaEeb0,2732
 sky/adaptors/oci.py,sha256=LfMSFUmkkNT6Yoz9FZHNl6UFSg4X1lJO4-x4ZbDdXTs,2831
 sky/adaptors/runpod.py,sha256=4Nt_BfZhJAKQNA3wO8cxvvNI8x4NsDGHu_4EhRDlGYQ,225
 sky/adaptors/vast.py,sha256=tpvmHi7IkQNzbbHVkeo04kUSajoEpSzXr2XgeO_I1LU,695
@@ -57,7 +57,7 @@ sky/clouds/gcp.py,sha256=0QpsI0Dso1xs3LhGlUq-Sq6WK-u11wN-57-vfcyhI5I,55154
 sky/clouds/ibm.py,sha256=R4JR96YfXstZ2B_IgFNVEX2SBAq3q0lSWz4y7FoFoeE,21474
 sky/clouds/kubernetes.py,sha256=7ki_zJZKnkhOPrHgVFq6azy5UhNKeeBOCSTjKCgj3vk,31709
 sky/clouds/lambda_cloud.py,sha256=ejqA_Wj5-325Y_QjQ__FY4HMO8sv_2tSRsufmaldcmI,12699
-sky/clouds/nebius.py,sha256=Z439DIaLCOqpCDBq8QhgiFyn5vIaZ7Rj7O6yZ1rjQ5g,12289
+sky/clouds/nebius.py,sha256=4180IruRMib7L9o60lrxrUDJtYhpX4lWFfAznbZoY6Q,12560
 sky/clouds/oci.py,sha256=irINbQsQ6YxRxGTMaCNsms3mZkIun2oJMMA1fMCRJyA,27072
 sky/clouds/paperspace.py,sha256=O7bH8YaHBLFuyj6rDz2bPDz_6OYWmNB9OLqnZH70yfY,10922
 sky/clouds/runpod.py,sha256=hzYB4td6qaged83xMAVKZ96bH40oZnrHXL7a_CKxXIw,11926
@@ -111,7 +111,7 @@ sky/jobs/constants.py,sha256=HrQbQA41Khor89ZoH112kjHRgehge-k4o0R9T9UbbaM,3088
 sky/jobs/controller.py,sha256=4G1CKI7M7D1BgJLbJMeqzg0iDDv7FR4ObB1BKZFFjhk,29585
 sky/jobs/recovery_strategy.py,sha256=RLrqq8B1likxTknPzt3_BqO26sFVpoatxzUuGfwc18A,26170
 sky/jobs/scheduler.py,sha256=IUW0a_69Pkvs4jqsWCXkeMDIZn-TTuPNyZvPLGRUYUM,12306
-sky/jobs/state.py,sha256=y9X1JGWpCokWBIwHZGB55fx39ZsilB1NZTbN_U8mPjA,43528
+sky/jobs/state.py,sha256=dIONpSuo5oolbCH1j7IWPlK97s5-NSl3whxH5Gn5DYI,43660
 sky/jobs/utils.py,sha256=SEeoXDWM-n2kg-eLd39kLWXbK7cavMst9s3xFVqaJg4,52580
 sky/jobs/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sky/jobs/client/sdk.py,sha256=mblI0hRH1It-I1hfs2_TdQTh_mtixcYJEZCWyKbqaaE,9707
@@ -175,7 +175,7 @@ sky/provision/lambda_cloud/lambda_utils.py,sha256=H2Qx4xdJyyEu2IXaj5AyppuPJW385n
 sky/provision/nebius/__init__.py,sha256=30I3181mu0W5g9fNvaWMPoBJZoGZ9RibuTpBH9P2pDg,558
 sky/provision/nebius/config.py,sha256=LK9kTDp2w6zZrn3vNdcSGgsgS-dL_j63Nh4_u3pqNiA,321
 sky/provision/nebius/instance.py,sha256=kfpZqwh56XV7h6XsX5dGJYilsVdRhC1EUgk0flaCo_E,11889
-sky/provision/nebius/utils.py,sha256=BxA7Hh6P7dKcXSw4tx2dyYd1txm-c7Psym3rRCAGLic,12785
+sky/provision/nebius/utils.py,sha256=o1Jr_pzH-2uk34obt9pSGXf2i2tljOo-uhXcryfFUxs,13145
 sky/provision/oci/__init__.py,sha256=5E6EUtTK3mqGVREw5TuVl5DxteBYTZigIii7c8gHExU,612
 sky/provision/oci/config.py,sha256=diSDTyHLokcuXGB2XgZCHFvsXa8bah1PP2XuMouW_UU,1650
 sky/provision/oci/instance.py,sha256=rVGee5y0qkoeLIP5vPDmLq1N8G1n8mru62VHb0Bv5iY,16784
@@ -344,9 +344,9 @@ sky/utils/kubernetes/k8s_gpu_labeler_setup.yaml,sha256=VLKT2KKimZu1GDg_4AIlIt488
 sky/utils/kubernetes/kubernetes_deploy_utils.py,sha256=iAjfyPclOs8qlALACcfxLpRAO9CZ-h16leFqXZ6tNaY,10096
 sky/utils/kubernetes/rsync_helper.sh,sha256=h4YwrPFf9727CACnMJvF3EyK_0OeOYKKt4su_daKekw,1256
 sky/utils/kubernetes/ssh_jump_lifecycle_manager.py,sha256=Kq1MDygF2IxFmu9FXpCxqucXLmeUrvs6OtRij6XTQbo,6554
-skypilot_nightly-1.0.0.dev20250224.dist-info/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
-skypilot_nightly-1.0.0.dev20250224.dist-info/METADATA,sha256=ooOoPMweqRG9pczdIUORjFZdEqb0bjxfyGkkHxBSgMg,19055
-skypilot_nightly-1.0.0.dev20250224.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-skypilot_nightly-1.0.0.dev20250224.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
-skypilot_nightly-1.0.0.dev20250224.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
-skypilot_nightly-1.0.0.dev20250224.dist-info/RECORD,,
+skypilot_nightly-1.0.0.dev20250225.dist-info/LICENSE,sha256=emRJAvE7ngL6x0RhQvlns5wJzGI3NEQ_WMjNmd9TZc4,12170
+skypilot_nightly-1.0.0.dev20250225.dist-info/METADATA,sha256=n0NjxuYEKzBIVHBricBiNx4ILxluNGwheNot34KQd2M,19055
+skypilot_nightly-1.0.0.dev20250225.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+skypilot_nightly-1.0.0.dev20250225.dist-info/entry_points.txt,sha256=StA6HYpuHj-Y61L2Ze-hK2IcLWgLZcML5gJu8cs6nU4,36
+skypilot_nightly-1.0.0.dev20250225.dist-info/top_level.txt,sha256=qA8QuiNNb6Y1OF-pCUtPEr6sLEwy2xJX06Bd_CrtrHY,4
+skypilot_nightly-1.0.0.dev20250225.dist-info/RECORD,,