konduktor-nightly 0.1.0.dev20251023104627__py3-none-any.whl → 0.1.0.dev20251024104753__py3-none-any.whl

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.dev20251024104753'
 __root_dir__ = os.path.dirname(os.path.abspath(__file__))

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/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/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/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/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.dist-info → konduktor_nightly-0.1.0.dev20251024104753.dist-info}/METADATA

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

{konduktor_nightly-0.1.0.dev20251023104627.dist-info → konduktor_nightly-0.1.0.dev20251024104753.dist-info}/RECORD

@@ -1,4 +1,4 @@
-konduktor/__init__.py,sha256=euAoDTHdzANHMdyIr3iKaAQpvyi4s5M_i560sLnWwcU,1574
+konduktor/__init__.py,sha256=FjWMNSOYV6pEAda12-HGCn8TZYDK0aAYzyFGQLTOB74,1574
 konduktor/adaptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 konduktor/adaptors/aws.py,sha256=s47Ra-GaqCQibzVfmD0pmwEWHif1EGO5opMbwkLxTCU,8244
 konduktor/adaptors/common.py,sha256=ZIqzjx77PIHUwpjfAQ1uX8B2aX78YMuGj4Bppd-MdyM,4183
@@ -10,10 +10,10 @@ konduktor/backends/constants.py,sha256=nt9G9AmFCOMwO4GuKgRQSzJJuKapOmaROp4_Y0tMF
 konduktor/backends/deployment.py,sha256=VXz0GMvFzqq8kd1-pXaNaTsEtSyMmVO2OSd-_-T9nIE,6171
 konduktor/backends/deployment_utils.py,sha256=rfVpiM0KlXma8SxA029TczrpbY-gRG3T0pj2eeARkE8,49257
 konduktor/backends/jobset.py,sha256=drt8Gc0iYQx18JWXBU6XfhUvC2xCKd8szSJ2JC4O20Q,8640
-konduktor/backends/jobset_utils.py,sha256=If4pv5peB_yXrJJwjkySgVzbjcxDEDWfkOQxUkwSlOk,26386
+konduktor/backends/jobset_utils.py,sha256=veY0O1Yn5YK9e3uZhBdeoATvKFjyARU9FYUzX5Ypyrc,26496
 konduktor/backends/pod_utils.py,sha256=KP_PAgsdNHFgt4Od-5gAtpifAKIL7DMBg7NJ44uqikg,14885
 konduktor/check.py,sha256=JennyWoaqSKhdyfUldd266KwVXTPJpcYQa4EED4a_BA,7569
-konduktor/cli.py,sha256=HOL_O7BpYCl7uF5mKkMvXLjFNMTeAh2hYkraaJ6zi3w,57674
+konduktor/cli.py,sha256=vb9SD8eJYjddiInVVQMq8K2BwqVDFoaMSTuaPhGp50I,57907
 konduktor/config.py,sha256=9upqgCCYvcu6fKw7tovEYC1MWTkAAir0_WHPdayylbI,15536
 konduktor/constants.py,sha256=T3AeXXxuQHINW_bAWyztvDeS8r4g8kXBGIwIq13cys0,1814
 konduktor/controller/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -75,9 +75,9 @@ konduktor/manifests/controller_deployment.yaml,sha256=6p3oSLkEVONZsvKZGqVop0Dhn4
 konduktor/manifests/dashboard_deployment.yaml,sha256=xJLd4FbPMAosI0fIv5_8y7dV9bw0Vsf81l-w4MB_aU8,2837
 konduktor/manifests/dmesg_daemonset.yaml,sha256=pSWt7YOeTYjS0l0iki1fvHOs7MhY-sH-RQfVW6JJyno,1391
 konduktor/manifests/pod_cleanup_controller.yaml,sha256=hziL1Ka1kCAEL9R7Tjvpb80iw1vcq9_3gwHCu75Bi0A,3939
-konduktor/resource.py,sha256=kfdhnUR_9kDLSm2sUAkv1sLQXyAkI08p3wIzCz1p7-M,20791
+konduktor/resource.py,sha256=JqEE3LZiBBd5vqAiHDk-nlLve_VUQHhgdo6BIgx2Xfk,21215
 konduktor/serving.py,sha256=4s8cQhsVjf-HByZF65pbMxuqaV319hUSQE9pC8gP4Sg,5405
-konduktor/task.py,sha256=LcTblvTNCSFNBHNUIOGN-i43whR8B6ccBm3HZTDoCMs,38050
+konduktor/task.py,sha256=FIWm_rC_63GPBoe-Hi8a_eJ0H8Szw747SwXYPrNtOWE,37820
 konduktor/templates/deployment.yaml.j2,sha256=ByA33oaBgWnSx_QZ6yKeaMcCjktphcwF8R5KiOlM4pE,4859
 konduktor/templates/jobset.yaml.j2,sha256=NQcVeRNsTLLmTnJRnkL1vr45mSeth-b11YShXn_RoSg,1323
 konduktor/templates/pod.yaml.j2,sha256=p9yE-AQkCF2Tgjd1QQiOLtzgI6Gbpps-MZYJZ9fQWIs,19159
@@ -99,9 +99,9 @@ konduktor/utils/rich_utils.py,sha256=ycADW6Ij3wX3uT8ou7T8qxX519RxlkJivsLvUahQaJo
 konduktor/utils/schemas.py,sha256=cr39nEAgjluhXoUYnvIwCwLBH8rLds37MBsF1uQv1rw,19067
 konduktor/utils/subprocess_utils.py,sha256=WoFkoFhGecPR8-rF8WJxbIe-YtV94LXz9UG64SDhCY4,9448
 konduktor/utils/ux_utils.py,sha256=LSH4b5lckD157qDF4keThxtkGdxNrAfGKmH1ewhZkm4,8646
-konduktor/utils/validator.py,sha256=gCB5v9Up9bCWD_92fS5ChfRRXj_m56Ky9uzd_77wXGI,16927
-konduktor_nightly-0.1.0.dev20251023104627.dist-info/LICENSE,sha256=MuuqTZbHvmqXR_aNKAXzggdV45ANd3wQ5YI7tnpZhm0,6586
-konduktor_nightly-0.1.0.dev20251023104627.dist-info/METADATA,sha256=mZ4Eqs59fnE0eMgtnmnOHv4N0p6hTnp_Hk8aAWd4kS4,4247
-konduktor_nightly-0.1.0.dev20251023104627.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-konduktor_nightly-0.1.0.dev20251023104627.dist-info/entry_points.txt,sha256=k3nG5wDFIJhNqsZWrHk4d0irIB2Ns9s47cjRWYsTCT8,48
-konduktor_nightly-0.1.0.dev20251023104627.dist-info/RECORD,,
+konduktor/utils/validator.py,sha256=UcLvZCk9Cpbbhw8r_ZJtTpMSTfY1NKqcyciKsPzRPZM,17222
+konduktor_nightly-0.1.0.dev20251024104753.dist-info/LICENSE,sha256=MuuqTZbHvmqXR_aNKAXzggdV45ANd3wQ5YI7tnpZhm0,6586
+konduktor_nightly-0.1.0.dev20251024104753.dist-info/METADATA,sha256=mXfjZjQzu3XmB8hxuhoM4rt_7ynOqileNOekKbwgY8M,4247
+konduktor_nightly-0.1.0.dev20251024104753.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+konduktor_nightly-0.1.0.dev20251024104753.dist-info/entry_points.txt,sha256=k3nG5wDFIJhNqsZWrHk4d0irIB2Ns9s47cjRWYsTCT8,48
+konduktor_nightly-0.1.0.dev20251024104753.dist-info/RECORD,,