PyPI - skypilot-nightly - Versions diffs - 1.0.0.dev20250720__py3-none-any.whl → 1.0.0.dev20250723__py3-none-any.whl - Mend

skypilot-nightly 1.0.0.dev20250720py3-none-any.whl → 1.0.0.dev20250723py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of skypilot-nightly might be problematic. Click here for more details.

Files changed (103) hide show

sky/dashboard/out/workspaces.html CHANGED Viewed

	@@ -1 +1 @@
1	- <!DOCTYPE html><html><head><meta charSet="utf-8"/><meta name="viewport" content="width=device-width"/><meta name="next-head-count" content="2"/><link rel="preload" href="/dashboard/_next/static/css/~~5122cb0a08486fd3~~.css" as="style"/><link rel="stylesheet" href="/dashboard/_next/static/css/~~5122cb0a08486fd3~~.css" data-n-g=""/><noscript data-n-css=""></noscript><script defer="" nomodule="" src="/dashboard/_next/static/chunks/polyfills-78c92fac7aa8fdd8.js"></script><script src="/dashboard/_next/static/chunks/webpack-~~26cdc782eed15a7d~~.js" defer=""></script><script src="/dashboard/_next/static/chunks/framework-efc06c2733009cd3.js" defer=""></script><script src="/dashboard/_next/static/chunks/main-c0a4f1ea606d48d2.js" defer=""></script><script src="/dashboard/_next/static/chunks/pages/_app-~~507712f30cd3cec3~~.js" defer=""></script><script src="/dashboard/_next/static/chunks/pages/workspaces-~~a1e43d9ef51a9cea~~.js" defer=""></script><script src="/dashboard/_next/static/~~pTQKG61ng32Zc7gsAROFJ~~/_buildManifest.js" defer=""></script><script src="/dashboard/_next/static/~~pTQKG61ng32Zc7gsAROFJ~~/_ssgManifest.js" defer=""></script></head><body><div id="__next"></div><script id="__NEXT_DATA__" type="application/json">{"props":{"pageProps":{}},"page":"/workspaces","query":{},"buildId":"~~pTQKG61ng32Zc7gsAROFJ~~","assetPrefix":"/dashboard","nextExport":true,"autoExport":true,"isFallback":false,"scriptLoader":[]}</script></body></html>
1	+ <!DOCTYPE html><html><head><meta charSet="utf-8"/><meta name="viewport" content="width=device-width"/><meta name="next-head-count" content="2"/><link rel="preload" href="/dashboard/_next/static/css/b3227360726f12eb.css" as="style"/><link rel="stylesheet" href="/dashboard/_next/static/css/b3227360726f12eb.css" data-n-g=""/><noscript data-n-css=""></noscript><script defer="" nomodule="" src="/dashboard/_next/static/chunks/polyfills-78c92fac7aa8fdd8.js"></script><script src="/dashboard/_next/static/chunks/webpack-a305898dc479711e.js" defer=""></script><script src="/dashboard/_next/static/chunks/framework-efc06c2733009cd3.js" defer=""></script><script src="/dashboard/_next/static/chunks/main-c0a4f1ea606d48d2.js" defer=""></script><script src="/dashboard/_next/static/chunks/pages/_app-da491665d4289aae.js" defer=""></script><script src="/dashboard/_next/static/chunks/pages/workspaces-5f7fe4b7d55b8612.js" defer=""></script><script src="/dashboard/_next/static/mym3Ciwp-zqU7ZpOLGnrW/_buildManifest.js" defer=""></script><script src="/dashboard/_next/static/mym3Ciwp-zqU7ZpOLGnrW/_ssgManifest.js" defer=""></script></head><body><div id="__next"></div><script id="__NEXT_DATA__" type="application/json">{"props":{"pageProps":{}},"page":"/workspaces","query":{},"buildId":"mym3Ciwp-zqU7ZpOLGnrW","assetPrefix":"/dashboard","nextExport":true,"autoExport":true,"isFallback":false,"scriptLoader":[]}</script></body></html>

sky/data/mounting_utils.py CHANGED Viewed

@@ -39,19 +39,32 @@ _GOOFYS_WRAPPER = ('$(if [ -S /dev/log ] ; then '
 def get_s3_mount_install_cmd() -> str:
-    """Returns a command to install S3 mount utility goofys."""
+    """Returns command for basic S3 mounting (goofys by default, rclone for
+    ARM64)."""
     # TODO(aylei): maintain our goofys fork under skypilot-org
-    install_cmd = ('ARCH=$(uname -m) && '
-                   'if [ "$ARCH" = "aarch64" ] || [ "$ARCH" = "arm64" ]; then '
-                   '  echo "goofys is not supported on $ARCH" && '
-                   f'  exit {exceptions.ARCH_NOT_SUPPORTED_EXIT_CODE}; '
-                   'else '
-                   '  ARCH_SUFFIX="amd64"; '
-                   'fi && '
-                   'sudo wget -nc https://github.com/aylei/goofys/'
-                   'releases/download/0.24.0-aylei-upstream/goofys '
-                   '-O /usr/local/bin/goofys && '
-                   'sudo chmod 755 /usr/local/bin/goofys')
+    install_cmd = (
+        'ARCH=$(uname -m) && '
+        'if [ "$ARCH" = "aarch64" ] || [ "$ARCH" = "arm64" ]; then '
+        # Use rclone for ARM64 since goofys doesn't support it
+        # Extract core rclone installation logic without redundant ARCH check
+        '  ARCH_SUFFIX="arm" && '
+        f'  (which dpkg > /dev/null 2>&1 && (which rclone > /dev/null || '
+        f'(cd ~ > /dev/null && curl -O https://downloads.rclone.org/'
+        f'{RCLONE_VERSION}/rclone-{RCLONE_VERSION}-linux-${{ARCH_SUFFIX}}.deb '
+        f'&& sudo dpkg -i rclone-{RCLONE_VERSION}-linux-${{ARCH_SUFFIX}}.deb '
+        f'&& rm -f rclone-{RCLONE_VERSION}-linux-${{ARCH_SUFFIX}}.deb))) || '
+        f'(which rclone > /dev/null || (cd ~ > /dev/null && curl -O '
+        f'https://downloads.rclone.org/{RCLONE_VERSION}/'
+        f'rclone-{RCLONE_VERSION}-linux-${{ARCH_SUFFIX}}.rpm && '
+        f'sudo yum --nogpgcheck install '
+        f'rclone-{RCLONE_VERSION}-linux-${{ARCH_SUFFIX}}.rpm -y && '
+        f'rm -f rclone-{RCLONE_VERSION}-linux-${{ARCH_SUFFIX}}.rpm)); '
+        'else '
+        '  sudo wget -nc https://github.com/aylei/goofys/'
+        'releases/download/0.24.0-aylei-upstream/goofys '
+        '-O /usr/local/bin/goofys && '
+        'sudo chmod 755 /usr/local/bin/goofys; '
+        'fi')
     return install_cmd
@@ -59,15 +72,30 @@ def get_s3_mount_install_cmd() -> str:
 def get_s3_mount_cmd(bucket_name: str,
                      mount_path: str,
                      _bucket_sub_path: Optional[str] = None) -> str:
-    """Returns a command to mount an S3 bucket using goofys."""
+    """Returns a command to mount an S3 bucket (goofys by default, rclone for
+    ARM64)"""
     if _bucket_sub_path is None:
         _bucket_sub_path = ''
     else:
         _bucket_sub_path = f':{_bucket_sub_path}'
-    mount_cmd = (f'{_GOOFYS_WRAPPER} -o allow_other '
-                 f'--stat-cache-ttl {_STAT_CACHE_TTL} '
-                 f'--type-cache-ttl {_TYPE_CACHE_TTL} '
-                 f'{bucket_name}{_bucket_sub_path} {mount_path}')
+    # Use rclone for ARM64 architectures since goofys doesn't support them
+    arch_check = 'ARCH=$(uname -m) && '
+    rclone_mount = (
+        f'{FUSERMOUNT3_SOFT_LINK_CMD} && '
+        f'rclone mount :s3:{bucket_name}{_bucket_sub_path} {mount_path} '
+        '--daemon --allow-other')
+    goofys_mount = (f'{_GOOFYS_WRAPPER} -o allow_other '
+                    f'--stat-cache-ttl {_STAT_CACHE_TTL} '
+                    f'--type-cache-ttl {_TYPE_CACHE_TTL} '
+                    f'{bucket_name}{_bucket_sub_path} {mount_path}')
+    mount_cmd = (f'{arch_check}'
+                 f'if [ "$ARCH" = "aarch64" ] || [ "$ARCH" = "arm64" ]; then '
+                 f'  {rclone_mount}; '
+                 f'else '
+                 f'  {goofys_mount}; '
+                 f'fi')
     return mount_cmd
@@ -76,17 +104,33 @@ def get_nebius_mount_cmd(nebius_profile_name: str,
                          endpoint_url: str,
                          mount_path: str,
                          _bucket_sub_path: Optional[str] = None) -> str:
-    """Returns a command to install Nebius mount utility goofys."""
+    """Returns a command to mount Nebius bucket (goofys by default, rclone for
+    ARM64)."""
     if _bucket_sub_path is None:
         _bucket_sub_path = ''
     else:
         _bucket_sub_path = f':{_bucket_sub_path}'
-    mount_cmd = (f'AWS_PROFILE={nebius_profile_name} {_GOOFYS_WRAPPER} '
-                 '-o allow_other '
-                 f'--stat-cache-ttl {_STAT_CACHE_TTL} '
-                 f'--type-cache-ttl {_TYPE_CACHE_TTL} '
-                 f'--endpoint {endpoint_url} '
-                 f'{bucket_name}{_bucket_sub_path} {mount_path}')
+    # Use rclone for ARM64 architectures since goofys doesn't support them
+    arch_check = 'ARCH=$(uname -m) && '
+    rclone_mount = (
+        f'{FUSERMOUNT3_SOFT_LINK_CMD} && '
+        f'AWS_PROFILE={nebius_profile_name} '
+        f'rclone mount :s3:{bucket_name}{_bucket_sub_path} {mount_path} '
+        f'--s3-endpoint {endpoint_url} --daemon --allow-other')
+    goofys_mount = (f'AWS_PROFILE={nebius_profile_name} {_GOOFYS_WRAPPER} '
+                    '-o allow_other '
+                    f'--stat-cache-ttl {_STAT_CACHE_TTL} '
+                    f'--type-cache-ttl {_TYPE_CACHE_TTL} '
+                    f'--endpoint {endpoint_url} '
+                    f'{bucket_name}{_bucket_sub_path} {mount_path}')
+    mount_cmd = (f'{arch_check}'
+                 f'if [ "$ARCH" = "aarch64" ] || [ "$ARCH" = "arm64" ]; then '
+                 f'  {rclone_mount}; '
+                 f'else '
+                 f'  {goofys_mount}; '
+                 f'fi')
     return mount_cmd
@@ -236,18 +280,35 @@ def get_r2_mount_cmd(r2_credentials_path: str,
                      bucket_name: str,
                      mount_path: str,
                      _bucket_sub_path: Optional[str] = None) -> str:
-    """Returns a command to install R2 mount utility goofys."""
+    """Returns a command to mount R2 bucket (goofys by default, rclone for
+    ARM64)."""
     if _bucket_sub_path is None:
         _bucket_sub_path = ''
     else:
         _bucket_sub_path = f':{_bucket_sub_path}'
-    mount_cmd = (f'AWS_SHARED_CREDENTIALS_FILE={r2_credentials_path} '
-                 f'AWS_PROFILE={r2_profile_name} {_GOOFYS_WRAPPER} '
-                 '-o allow_other '
-                 f'--stat-cache-ttl {_STAT_CACHE_TTL} '
-                 f'--type-cache-ttl {_TYPE_CACHE_TTL} '
-                 f'--endpoint {endpoint_url} '
-                 f'{bucket_name}{_bucket_sub_path} {mount_path}')
+    # Use rclone for ARM64 architectures since goofys doesn't support them
+    arch_check = 'ARCH=$(uname -m) && '
+    rclone_mount = (
+        f'{FUSERMOUNT3_SOFT_LINK_CMD} && '
+        f'AWS_SHARED_CREDENTIALS_FILE={r2_credentials_path} '
+        f'AWS_PROFILE={r2_profile_name} '
+        f'rclone mount :s3:{bucket_name}{_bucket_sub_path} {mount_path} '
+        f'--s3-endpoint {endpoint_url} --daemon --allow-other')
+    goofys_mount = (f'AWS_SHARED_CREDENTIALS_FILE={r2_credentials_path} '
+                    f'AWS_PROFILE={r2_profile_name} {_GOOFYS_WRAPPER} '
+                    '-o allow_other '
+                    f'--stat-cache-ttl {_STAT_CACHE_TTL} '
+                    f'--type-cache-ttl {_TYPE_CACHE_TTL} '
+                    f'--endpoint {endpoint_url} '
+                    f'{bucket_name}{_bucket_sub_path} {mount_path}')
+    mount_cmd = (f'{arch_check}'
+                 f'if [ "$ARCH" = "aarch64" ] || [ "$ARCH" = "arm64" ]; then '
+                 f'  {rclone_mount}; '
+                 f'else '
+                 f'  {goofys_mount}; '
+                 f'fi')
     return mount_cmd

sky/global_user_state.py CHANGED Viewed

@@ -17,7 +17,6 @@ import typing
 from typing import Any, Dict, List, Optional, Set, Tuple
 import uuid
-from alembic import command as alembic_command
 import sqlalchemy
 from sqlalchemy import exc as sqlalchemy_exc
 from sqlalchemy import orm
@@ -243,8 +242,8 @@ def create_table(engine: sqlalchemy.engine.Engine):
         engine, migration_utils.GLOBAL_USER_STATE_DB_NAME)
     # pylint: disable=line-too-long
     alembic_config.config_ini_section = migration_utils.GLOBAL_USER_STATE_DB_NAME
-    alembic_command.upgrade(alembic_config,
-                            migration_utils.GLOBAL_USER_STATE_VERSION)
+    migration_utils.safe_alembic_upgrade(
+        engine, alembic_config, migration_utils.GLOBAL_USER_STATE_VERSION)
 def initialize_and_get_db() -> sqlalchemy.engine.Engine:

sky/jobs/state.py CHANGED Viewed

@@ -10,7 +10,6 @@ import time
 import typing
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
-from alembic import command as alembic_command
 import colorama
 import sqlalchemy
 from sqlalchemy import exc as sqlalchemy_exc
@@ -134,7 +133,8 @@ def create_table(engine: sqlalchemy.engine.Engine):
     alembic_config = migration_utils.get_alembic_config(
         engine, migration_utils.SPOT_JOBS_DB_NAME)
     alembic_config.config_ini_section = migration_utils.SPOT_JOBS_DB_NAME
-    alembic_command.upgrade(alembic_config, migration_utils.SPOT_JOBS_VERSION)
+    migration_utils.safe_alembic_upgrade(engine, alembic_config,
+                                         migration_utils.SPOT_JOBS_VERSION)
 def initialize_and_get_db() -> sqlalchemy.engine.Engine:

sky/provision/nebius/utils.py CHANGED Viewed

@@ -41,10 +41,7 @@ def get_project_by_region(region: str) -> str:
     #  Check is there project if in config
     project_id = skypilot_config.get_effective_region_config(
-        cloud='nebius',
-        region=None,
-        keys=(region, 'project_id'),
-        default_value=None)
+        cloud='nebius', region=region, keys=('project_id',), default_value=None)
     if project_id is not None:
         return project_id
     for project in projects.items:
@@ -189,8 +186,8 @@ def launch(cluster_name_on_cloud: str,
         if preset == '8gpu-128vcpu-1600gb':
             fabric = skypilot_config.get_effective_region_config(
                 cloud='nebius',
-                region=None,
-                keys=(region, 'fabric'),
+                region=region,
+                keys=('fabric',),
                 default_value=None)
             # Auto-select fabric if network_tier=best and no fabric configured

sky/server/common.py CHANGED Viewed

@@ -252,8 +252,9 @@ def get_dashboard_url(server_url: str,
 @annotations.lru_cache(scope='global')
-def is_api_server_local():
-    return get_server_url() in AVAILABLE_LOCAL_API_SERVER_URLS
+def is_api_server_local(endpoint: Optional[str] = None):
+    server_url = endpoint if endpoint is not None else get_server_url()
+    return server_url in AVAILABLE_LOCAL_API_SERVER_URLS
 def _handle_non_200_server_status(
@@ -566,7 +567,7 @@ def check_server_healthy(
     api_server_status = api_server_info.status
     if api_server_status == ApiServerStatus.VERSION_MISMATCH:
         msg = api_server_info.error
-        if is_api_server_local():
+        if is_api_server_local(endpoint):
             # For local server, just hint user to restart the server to get
             # a consistent version.
             msg = _LOCAL_API_SERVER_RESTART_HINT

sky/setup_files/MANIFEST.in CHANGED Viewed

@@ -18,5 +18,5 @@ include sky/server/html/*
 recursive-include sky/dashboard/out *
 include sky/users/*.conf
 include sky/utils/*.sh
-include alembic.ini
+include sky/setup_files/alembic.ini
 recursive-include sky/schemas/db *

sky/setup_files/alembic.ini CHANGED Viewed

@@ -94,10 +94,6 @@ version_table = alembic_version_state_db
 version_locations = %(here)s/../schemas/db/spot_jobs
 version_table = alembic_version_spot_jobs_db
-[sky_config_db]
-version_locations = %(here)s/../schemas/db/skypilot_config
-version_table = alembic_version_sky_config_db
 [post_write_hooks]
 # post_write_hooks defines scripts or Python functions that are run
 # on newly generated revision scripts.  See the documentation for further

sky/skylet/constants.py CHANGED Viewed

@@ -1,4 +1,5 @@
 """Constants for SkyPilot."""
+import os
 from typing import List, Tuple
 from packaging import version
@@ -491,3 +492,6 @@ DEFAULT_PRIORITY = 0
 GRACE_PERIOD_SECONDS_ENV_VAR = SKYPILOT_ENV_VAR_PREFIX + 'GRACE_PERIOD_SECONDS'
 COST_REPORT_DEFAULT_DAYS = 30
+# The directory for file locks.
+SKY_LOCKS_DIR = os.path.expanduser('~/.sky/locks')

sky/skypilot_config.py CHANGED Viewed

@@ -58,10 +58,8 @@ import threading
 import typing
 from typing import Any, Dict, Iterator, List, Optional, Tuple, Union
-from alembic import command as alembic_command
 import filelock
 import sqlalchemy
-from sqlalchemy import exc as sqlalchemy_exc
 from sqlalchemy import orm
 from sqlalchemy.dialects import postgresql
 from sqlalchemy.dialects import sqlite
@@ -78,7 +76,6 @@ from sky.utils import context
 from sky.utils import schemas
 from sky.utils import ux_utils
 from sky.utils.db import db_utils
-from sky.utils.db import migration_utils
 from sky.utils.kubernetes import config_map_utils
 if typing.TYPE_CHECKING:
@@ -574,17 +571,11 @@ def _reload_config_as_server() -> None:
             'if db config is specified, no other config is allowed')
     if db_url:
-        with migration_utils.db_lock(migration_utils.SKYPILOT_CONFIG_DB_NAME):
+        with _DB_USE_LOCK:
             sqlalchemy_engine = sqlalchemy.create_engine(db_url,
                                                          poolclass=NullPool)
-            # Get alembic config for sky config db and run migrations
-            alembic_config = migration_utils.get_alembic_config(
-                sqlalchemy_engine, migration_utils.SKYPILOT_CONFIG_DB_NAME)
-            # pylint: disable=line-too-long
-            alembic_config.config_ini_section = migration_utils.SKYPILOT_CONFIG_DB_NAME
-            alembic_command.upgrade(alembic_config,
-                                    migration_utils.SKYPILOT_CONFIG_VERSION)
+            db_utils.add_tables_to_db_sqlalchemy(Base.metadata,
+                                                 sqlalchemy_engine)
             def _get_config_yaml_from_db(
                     key: str) -> Optional[config_utils.Config]:
@@ -872,25 +863,8 @@ def update_api_server_config_no_lock(config: config_utils.Config) -> None:
             with _DB_USE_LOCK:
                 sqlalchemy_engine = sqlalchemy.create_engine(existing_db_url,
                                                              poolclass=NullPool)
-                # Get alembic config for sky config db and run migrations
-                alembic_config = migration_utils.get_alembic_config(
-                    sqlalchemy_engine, 'sky_config_db')
-                alembic_config.config_ini_section = 'sky_config_db'
-                try:
-                    alembic_command.upgrade(alembic_config, '001')
-                except (sqlalchemy_exc.IntegrityError,
-                        sqlalchemy_exc.OperationalError) as e:
-                    # If the version already exists (due to concurrent
-                    # initialization), we can safely ignore this error
-                    if ('UNIQUE constraint failed: '
-                            'alembic_version_sky_config_db.version_num'
-                            in str(e) or
-                            'table alembic_version_sky_config_db already exists'
-                            in str(e)):
-                        pass
-                    else:
-                        raise
+                db_utils.add_tables_to_db_sqlalchemy(Base.metadata,
+                                                     sqlalchemy_engine)
                 def _set_config_yaml_to_db(key: str,
                                            config: config_utils.Config):

sky/utils/common_utils.py CHANGED Viewed

@@ -562,8 +562,9 @@ def read_yaml_all(path: str) -> List[Dict[str, Any]]:
         return read_yaml_all_str(f.read())
-def dump_yaml(path: str, config: Union[List[Dict[str, Any]],
-                                       Dict[str, Any]]) -> None:
+def dump_yaml(path: str,
+              config: Union[List[Dict[str, Any]], Dict[str, Any]],
+              blank: bool = False) -> None:
     """Dumps a YAML file.
     Args:
@@ -571,7 +572,11 @@ def dump_yaml(path: str, config: Union[List[Dict[str, Any]],
         config: the configuration to dump.
     """
     with open(path, 'w', encoding='utf-8') as f:
-        f.write(dump_yaml_str(config))
+        contents = dump_yaml_str(config)
+        if blank and isinstance(config, dict) and len(config) == 0:
+            # when dumping to yaml, an empty dict will go in as {}.
+            contents = ''
+        f.write(contents)
 def dump_yaml_str(config: Union[List[Dict[str, Any]], Dict[str, Any]]) -> str:

sky/utils/config_utils.py CHANGED Viewed

@@ -248,6 +248,8 @@ def get_cloud_config_value_from_dict(
     region_key = None
     if cloud == 'kubernetes':
         region_key = 'context_configs'
+    if cloud == 'nebius':
+        region_key = 'region_configs'
     per_context_config = None
     if region is not None and region_key is not None:
@@ -255,6 +257,19 @@ def get_cloud_config_value_from_dict(
             keys=(cloud, region_key, region) + keys,
             default_value=None,
             override_configs=override_configs)
+        if not per_context_config and cloud == 'nebius':
+            # TODO (kyuds): Backward compatibility, remove after 0.11.0.
+            per_context_config = input_config.get_nested(
+                keys=(cloud, region) + keys,
+                default_value=None,
+                override_configs=override_configs)
+            if per_context_config is not None:
+                logger.info(
+                    'Nebius configuration is using the legacy format. \n'
+                    'This format will be deprecated after 0.11.0, refer to '
+                    '`https://docs.skypilot.co/en/latest/reference/config.html#nebius` '  # pylint: disable=line-too-long
+                    'for the new format. Please use `region_configs` to specify region specific configuration.'
+                )
     # if no override found for specified region
     general_config = input_config.get_nested(keys=(cloud,) + keys,
                                              default_value=default_value,

sky/utils/db/migration_utils.py CHANGED Viewed

@@ -1,9 +1,12 @@
 """Constants for the database schemas."""
 import contextlib
+import logging
 import os
+from alembic import command as alembic_command
 from alembic.config import Config
+from alembic.runtime import migration
 import filelock
 import sqlalchemy
@@ -13,10 +16,6 @@ GLOBAL_USER_STATE_DB_NAME = 'state_db'
 GLOBAL_USER_STATE_VERSION = '001'
 GLOBAL_USER_STATE_LOCK_PATH = '~/.sky/locks/.state_db.lock'
-SKYPILOT_CONFIG_DB_NAME = 'skypilot_config_db'
-SKYPILOT_CONFIG_VERSION = '001'
-SKYPILOT_CONFIG_LOCK_PATH = '~/.sky/locks/.skypilot_config_db.lock'
 SPOT_JOBS_DB_NAME = 'spot_jobs_db'
 SPOT_JOBS_VERSION = '001'
 SPOT_JOBS_LOCK_PATH = '~/.sky/locks/.spot_jobs_db.lock'
@@ -51,3 +50,44 @@ def get_alembic_config(engine: sqlalchemy.engine.Engine, section: str):
     alembic_cfg.set_section_option(section, 'sqlalchemy.url', url)
     return alembic_cfg
+def safe_alembic_upgrade(engine: sqlalchemy.engine.Engine,
+                         alembic_config: Config, target_revision: str):
+    """Only upgrade if current version is older than target.
+    This handles the case where a database was created with a newer version of
+    the code and we're now running older code. Since our migrations are purely
+    additive, it's safe to run a newer database with older code.
+    Args:
+        engine: SQLAlchemy engine for the database
+        alembic_config: Alembic configuration object
+        target_revision: Target revision to upgrade to (e.g., '001')
+    """
+    # set alembic logger to warning level
+    alembic_logger = logging.getLogger('alembic')
+    alembic_logger.setLevel(logging.WARNING)
+    current_rev = None
+    # Get the current revision from the database
+    version_table = alembic_config.get_section_option(
+        alembic_config.config_ini_section, 'version_table', 'alembic_version')
+    with engine.connect() as connection:
+        context = migration.MigrationContext.configure(
+            connection, opts={'version_table': version_table})
+        current_rev = context.get_current_revision()
+    if current_rev is None:
+        alembic_command.upgrade(alembic_config, target_revision)
+        return
+    # Compare revisions - assuming they are numeric strings like '001', '002'
+    current_rev_num = int(current_rev)
+    target_rev_num = int(target_revision)
+    # only upgrade if current revision is older than target revision
+    if current_rev_num < target_rev_num:
+        alembic_command.upgrade(alembic_config, target_revision)

skypilot-nightly 1.0.0.dev20250720__py3-none-any.whl → 1.0.0.dev20250723__py3-none-any.whl

Potentially problematic release.

skypilot-nightly 1.0.0.dev20250720py3-none-any.whl → 1.0.0.dev20250723py3-none-any.whl