skypilot-nightly 1.0.0.dev20250224__tar.gz → 1.0.0.dev20250225__tar.gz

{skypilot_nightly-1.0.0.dev20250224/skypilot_nightly.egg-info → skypilot_nightly-1.0.0.dev20250225}/PKG-INFO

@@ -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 → skypilot_nightly-1.0.0.dev20250225}/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__))
 
 

{skypilot_nightly-1.0.0.dev20250224 → skypilot_nightly-1.0.0.dev20250225}/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:

{skypilot_nightly-1.0.0.dev20250224 → skypilot_nightly-1.0.0.dev20250225}/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}'

{skypilot_nightly-1.0.0.dev20250224 → skypilot_nightly-1.0.0.dev20250225}/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('

{skypilot_nightly-1.0.0.dev20250224 → skypilot_nightly-1.0.0.dev20250225}/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.)

{skypilot_nightly-1.0.0.dev20250224 → skypilot_nightly-1.0.0.dev20250225}/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 → skypilot_nightly-1.0.0.dev20250225/skypilot_nightly.egg-info}/PKG-INFO

@@ -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