konduktor-nightly 0.1.0.dev20251023104627__tar.gz → 0.1.0.dev20251025104336__tar.gz

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: konduktor-nightly
-Version: 0.1.0.dev20251023104627
+Version: 0.1.0.dev20251025104336
 Summary: GPU Cluster Health Management
 Author: Andrew Aikawa
 Author-email: asai@berkeley.edu

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/konduktor/__init__.py

@@ -11,7 +11,7 @@ from konduktor.task import Task
 __all__ = ['launch', 'Resources', 'Task', 'Serving']
 
 # Replaced with the current commit when building the wheels.
-_KONDUKTOR_COMMIT_SHA = '77ea1a3b26250e8ce18a507a4ab0d2538e351cb5'
+_KONDUKTOR_COMMIT_SHA = '431cdb62b9ab33df94d918db9d957bc4c54c2096'
 os.makedirs(os.path.expanduser('~/.konduktor'), exist_ok=True)
 
 
@@ -45,5 +45,5 @@ def _get_git_commit():
 
 
 __commit__ = _get_git_commit()
-__version__ = '1.0.0.dev0.1.0.dev20251023104627'
+__version__ = '1.0.0.dev0.1.0.dev20251025104336'
 __root_dir__ = os.path.dirname(os.path.abspath(__file__))

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/konduktor/backends/jobset_utils.py

@@ -5,15 +5,12 @@ import json
 import tempfile
 import time
 import typing
-from datetime import datetime, timezone
+from datetime import datetime, timedelta, timezone
 from typing import Any, Dict, Optional, Tuple
 
 import click
 import colorama
 
-if typing.TYPE_CHECKING:
-    from datetime import timedelta
-
 import konduktor
 from konduktor import kube_client, logging
 from konduktor.backends import constants as backend_constants
@@ -428,7 +425,9 @@ def _parse_timestamp_filter(timestamp_str: str) -> datetime:
                         seconds=abs(local_offset)
                     )
                 else:
-                    dt = dt.replace(tzinfo=timezone.utc)
+                    # Handle date-only format (local midnight --> UTC)
+                    local_tz = datetime.now().astimezone().tzinfo
+                    return dt.replace(tzinfo=local_tz).astimezone(timezone.utc)
             return dt
         except ValueError:
             continue

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/konduktor/cli.py

@@ -161,7 +161,9 @@ def _make_task_with_overrides(
     if workdir is not None:
         task.workdir = workdir
 
-    task.set_resources_override(override_params)
+    # perform overrides from CLI
+    if override_params:
+        task.set_resources_override(override_params)
     if task.serving:
         task.set_serving_override(serving_override_params)
 
@@ -653,28 +655,23 @@ def status(
     all_users: bool, limit: Optional[int], after: Optional[str], before: Optional[str]
 ):
     # NOTE(dev): Keep the docstring consistent between the Python API and CLI.
-    """Shows list of all the jobs with optional filtering and pagination
-
-    Args:
-        all_users (bool): whether to show all jobs for all users
-        limit (Optional[int]): maximum number of jobs to display
-        after (Optional[str]): show jobs created after this timestamp
-        before (Optional[str]): show jobs created before this timestamp
+    """Shows list of all the jobs with optional filtering and pagination.
 
+    \b
     Examples:
-        konduktor status --limit 10
-        konduktor status --before "08/06/25 03:53PM"
-        konduktor status --all-users --limit 10 --after "08/06/25 03:53PM"
-
-    Note:
-        When using --before or --after timestamps, passing in "08/06/25" is
-        equivalent to passing in "08/06/25 00:00".
-        When using --before or --after timestamps, passing in "03:53PM" is
-        equivalent to passing in "03:53:00PM".
-        Timestamps shown in "konduktor startus" are truncated and are in the
-        local timezone. ex. "03:53:55PM" --> "03:53PM"
-        and would show up in --after "03:53PM" but not in --before "03:53PM"
-        despite status showing as "03:53PM".
+      konduktor status --limit 10
+      konduktor status --before "08/06/25 03:53PM"
+      konduktor status --all-users --limit 10 --after "08/06/25 03:53PM"
+
+    \b
+    Notes:
+      • When using --before or --after timestamps, "08/06/25"
+        is equivalent to "08/06/25 00:00".
+      • "03:53PM" is equivalent to "03:53:00PM".
+      • Timestamps shown in "konduktor status" are truncated
+        and are in the local timezone.
+        Example: "03:53:55PM" → "03:53PM" — would show up in
+        --after "03:53PM" but not in --before "03:53PM".
     """
     context = kubernetes_utils.get_current_kube_config_context_name()
     namespace = kubernetes_utils.get_kube_config_context_namespace(context)
@@ -802,6 +799,13 @@ def logs(
     # pylint: disable=bad-docstring-quotes
     help='Skip confirmation prompt.',
 )
+@click.option(
+    '--skip-image-check',
+    '-s',
+    is_flag=True,
+    default=False,
+    help='Skip Docker image validation checks for faster startup.',
+)
 def launch(
     entrypoint: Tuple[str, ...],
     dryrun: bool,
@@ -820,6 +824,7 @@ def launch(
     env: List[Tuple[str, str]],
     disk_size: Optional[int],
     yes: bool,
+    skip_image_check: bool,
 ):
     """Launch a task.
 
@@ -829,6 +834,9 @@ def launch(
     # NOTE(dev): Keep the docstring consistent between the Python API and CLI.
     env = _merge_env_vars(env_file, env)
 
+    if skip_image_check:
+        os.environ['KONDUKTOR_SKIP_IMAGE_CHECK'] = '1'
+
     task = _make_task_with_overrides(
         entrypoint=entrypoint,
         name=name,
@@ -1694,6 +1702,13 @@ def serve():
     # pylint: disable=bad-docstring-quotes
     help='Skip confirmation prompt.',
 )
+@click.option(
+    '--skip-image-check',
+    '-s',
+    is_flag=True,
+    default=False,
+    help='Skip Docker image validation checks for faster startup.',
+)
 def serve_launch(
     entrypoint: Tuple[str, ...],
     dryrun: bool,
@@ -1716,6 +1731,7 @@ def serve_launch(
     ports: Optional[int],
     probe: Optional[str],
     yes: bool,
+    skip_image_check: bool = False,
 ):
     """Launch a deployment to serve.
 
@@ -1725,6 +1741,9 @@ def serve_launch(
     # NOTE(dev): Keep the docstring consistent between the Python API and CLI.
     env = _merge_env_vars(env_file, env)
 
+    if skip_image_check:
+        os.environ['KONDUKTOR_SKIP_IMAGE_CHECK'] = '1'
+
     task = _make_task_with_overrides(
         entrypoint=entrypoint,
         name=name,
@@ -1739,6 +1758,7 @@ def serve_launch(
         image_id=image_id,
         env=env,
         disk_size=disk_size,
+        # serving stuff
         min_replicas=min_replicas,
         max_replicas=max_replicas,
         ports=ports,

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/konduktor/resource.py

@@ -59,6 +59,8 @@ class Resources:
         # Internal use only.
         # pylint: disable=invalid-name
         _cluster_config_overrides: Optional[Dict[str, Any]] = None,
+        # used to prevent double validation of image (would happen from overrides)
+        _validate_image: bool = True,
     ):
         """Initialize a Resources object.
 
@@ -124,7 +126,8 @@ class Resources:
         if isinstance(image_id, str):
             self._image_id = image_id.strip()
             # Validate Docker image format and existence
-            validator.validate_and_warn_image(self._image_id, 'task')
+            if _validate_image:
+                validator.validate_and_warn_image(self._image_id, 'task')
 
         self._labels = labels
         self._cluster_config_overrides = _cluster_config_overrides
@@ -435,15 +438,19 @@ class Resources:
 
     def copy(self, **override) -> 'Resources':
         """Returns a copy of the given Resources."""
+        # used to prevent double validation of image (would happen from overrides)
+        new_image_id = override.pop('image_id', self.image_id)
         resources = Resources(
             cloud=override.pop('cloud', self.cloud),
             cpus=override.pop('cpus', self._cpus),
             memory=override.pop('memory', self.memory),
             accelerators=override.pop('accelerators', self.accelerators),
             disk_size=override.pop('disk_size', self.disk_size),
-            image_id=override.pop('image_id', self.image_id),
+            image_id=new_image_id,
             labels=override.pop('labels', self.labels),
             job_config=override.pop('job_config', self.job_config),
+            # used to prevent double validation of image (would happen from overrides)
+            _validate_image=(new_image_id != self.image_id),
         )
         assert len(override) == 0
         return resources

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/konduktor/task.py

@@ -29,7 +29,7 @@ import konduktor
 from konduktor import constants, logging
 from konduktor.data import data_utils
 from konduktor.data import storage as storage_lib
-from konduktor.utils import common_utils, exceptions, schemas, ux_utils, validator
+from konduktor.utils import common_utils, exceptions, schemas, ux_utils
 
 logger = logging.get_logger(__name__)
 
@@ -388,10 +388,6 @@ class Task:
             )
             resources_config['_cluster_config_overrides'] = cluster_config_override
 
-        # Validate Docker image if specified in resources
-        if 'image_id' in resources_config and resources_config['image_id']:
-            validator.validate_and_warn_image(resources_config['image_id'], 'task')
-
         task.set_resources(konduktor.Resources.from_yaml_config(resources_config))
 
         # Parse serving field.

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/konduktor/utils/validator.py

@@ -37,6 +37,11 @@ SCHEMA_URLS = {
 logger = logging.get_logger(__name__)
 
 
+def _skip_image_checks() -> bool:
+    val = os.getenv('KONDUKTOR_SKIP_IMAGE_CHECK', '')
+    return val.lower() in ('1', 'true', 'yes', 'y')
+
+
 def case_insensitive_enum(validator, enums, instance, schema):
     del validator, schema  # Unused.
     if instance.lower() not in [enum.lower() for enum in enums]:
@@ -419,6 +424,13 @@ def validate_and_warn_image(image_id: str, context: str = 'task') -> None:
     if not image_id:
         return
 
+    if _skip_image_checks():
+        logger.info(
+            'Skipping Docker image validation for %s',
+            image_id,
+        )
+        return
+
     status, message = validate_docker_image(image_id)
 
     if status == 'invalid':

{konduktor_nightly-0.1.0.dev20251023104627 → konduktor_nightly-0.1.0.dev20251025104336}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "konduktor-nightly"
-version = "0.1.0.dev20251023104627"
+version = "0.1.0.dev20251025104336"
 description = "GPU Cluster Health Management"
 packages = [
     {include = "konduktor"}