zenml-nightly 0.82.0.dev20250506__py3-none-any.whl → 0.82.0.dev20250508__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.82.0.dev20250506
+0.82.0.dev20250508

zenml/actions/pipeline_run/pipeline_run_action.py

@@ -101,6 +101,7 @@ class PipelineRunActionHandler(BaseActionHandler):
             template=template,
             run_config=config.run_config,
             auth_context=auth_context,
+            sync=True,
         )
 
     def _validate_configuration(

zenml/config/server_config.py

@@ -40,6 +40,7 @@ from zenml.constants import (
     DEFAULT_ZENML_SERVER_GENERIC_API_TOKEN_MAX_LIFETIME,
     DEFAULT_ZENML_SERVER_LOGIN_RATE_LIMIT_DAY,
     DEFAULT_ZENML_SERVER_LOGIN_RATE_LIMIT_MINUTE,
+    DEFAULT_ZENML_SERVER_MAX_CONCURRENT_TEMPLATE_RUNS,
     DEFAULT_ZENML_SERVER_MAX_DEVICE_AUTH_ATTEMPTS,
     DEFAULT_ZENML_SERVER_MAX_REQUEST_BODY_SIZE_IN_BYTES,
     DEFAULT_ZENML_SERVER_NAME,
@@ -153,6 +154,8 @@ class ServerConfiguration(BaseModel):
             server.
         workload_manager_implementation_source: Source pointing to a class
             implementing the workload management interface.
+        max_concurrent_template_runs: The maximum number of concurrent template
+            runs that can be executed on the server.
         pipeline_run_auth_window: The default time window in minutes for which
             a pipeline run action is allowed to authenticate with the ZenML
             server.
@@ -291,6 +294,9 @@ class ServerConfiguration(BaseModel):
     feature_gate_implementation_source: Optional[str] = None
     reportable_resources: List[str] = []
     workload_manager_implementation_source: Optional[str] = None
+    max_concurrent_template_runs: int = (
+        DEFAULT_ZENML_SERVER_MAX_CONCURRENT_TEMPLATE_RUNS
+    )
     pipeline_run_auth_window: int = (
         DEFAULT_ZENML_SERVER_PIPELINE_RUN_AUTH_WINDOW
     )

zenml/constants.py

@@ -291,6 +291,7 @@ DEFAULT_ZENML_SERVER_GENERIC_API_TOKEN_MAX_LIFETIME = (
     60 * 60 * 24 * 7
 )  # 7 days
 DEFAULT_ZENML_SERVER_FILE_DOWNLOAD_SIZE_LIMIT = 2 * 1024 * 1024 * 1024  # 20 GB
+DEFAULT_ZENML_SERVER_MAX_CONCURRENT_TEMPLATE_RUNS = 2
 
 DEFAULT_ZENML_SERVER_SECURE_HEADERS_HSTS = (
     "max-age=63072000; includeSubdomains"

zenml/exceptions.py

@@ -216,3 +216,7 @@ class BackupSecretsStoreNotConfiguredError(NotImplementedError):
 
 class CustomFlavorImportError(ImportError):
     """Raised when failing to import a custom flavor."""
+
+
+class MaxConcurrentTasksError(ZenMLBaseException):
+    """Raised when the maximum number of concurrent tasks is reached."""

zenml/utils/docker_utils.py

@@ -266,12 +266,21 @@ def push_image(
     logger.info("Finished pushing Docker image.")
 
     image_name_without_tag, _ = image_name.rsplit(":", maxsplit=1)
+    prefix_candidates = [f"{image_name_without_tag}@"]
+
+    if image_name_without_tag.startswith(("index.docker.io/", "docker.io/")):
+        # When looking for the repo digest later, Docker sometimes removes the
+        # index prefix, so we make sure to check for a digest with and without.
+        image_name_without_index = image_name_without_tag.split(
+            "/", maxsplit=1
+        )[1]
+        prefix_candidates.append(f"{image_name_without_index}@")
 
     image = docker_client.images.get(image_name)
     repo_digests: List[str] = image.attrs["RepoDigests"]
 
     for digest in repo_digests:
-        if digest.startswith(f"{image_name_without_tag}@"):
+        if digest.startswith(tuple(prefix_candidates)):
             return digest
 
     for info in reversed(aux_info):

zenml/zen_server/exceptions.py

@@ -25,6 +25,7 @@ from zenml.exceptions import (
     EntityCreationError,
     EntityExistsError,
     IllegalOperationError,
+    MaxConcurrentTasksError,
     MethodNotAllowedError,
     SubscriptionUpgradeRequiredError,
     ValidationError,
@@ -85,6 +86,8 @@ REST_API_EXCEPTIONS: List[Tuple[Type[Exception], int]] = [
     (ValueError, 400),
     # 422 Unprocessable Entity
     (ValueError, 422),
+    # 429 Too Many Requests
+    (MaxConcurrentTasksError, 429),
     # 500 Internal Server Error
     (EntityCreationError, 500),
     (RuntimeError, 500),

zenml/zen_server/routers/run_templates_endpoints.py

@@ -16,7 +16,11 @@
 from typing import Optional, Union
 from uuid import UUID
 
-from fastapi import APIRouter, BackgroundTasks, Depends, Security
+from fastapi import (
+    APIRouter,
+    Depends,
+    Security,
+)
 
 from zenml.analytics.enums import AnalyticsEvent
 from zenml.analytics.utils import track_handler
@@ -220,12 +224,12 @@ if server_config().workload_manager_enabled:
             401: error_response,
             404: error_response,
             422: error_response,
+            429: error_response,
         },
     )
     @handle_exceptions
     def create_template_run(
         template_id: UUID,
-        background_tasks: BackgroundTasks,
         config: Optional[PipelineRunConfiguration] = None,
         auth_context: AuthContext = Security(authorize),
     ) -> PipelineRunResponse:
@@ -233,14 +237,15 @@ if server_config().workload_manager_enabled:
 
         Args:
             template_id: The ID of the template.
-            background_tasks: Background tasks.
             config: Configuration for the pipeline run.
             auth_context: Authentication context.
 
         Returns:
             The created pipeline run.
         """
-        from zenml.zen_server.template_execution.utils import run_template
+        from zenml.zen_server.template_execution.utils import (
+            run_template,
+        )
 
         with track_handler(
             event=AnalyticsEvent.EXECUTED_RUN_TEMPLATE,
@@ -268,6 +273,5 @@ if server_config().workload_manager_enabled:
             return run_template(
                 template=template,
                 auth_context=auth_context,
-                background_tasks=background_tasks,
                 run_config=config,
             )

zenml/zen_server/template_execution/utils.py

@@ -3,10 +3,11 @@
 import copy
 import hashlib
 import sys
-from typing import Any, Dict, List, Optional
+import threading
+from concurrent.futures import Future, ThreadPoolExecutor
+from typing import Any, Callable, Dict, List, Optional
 from uuid import UUID
 
-from fastapi import BackgroundTasks
 from packaging import version
 
 from zenml.analytics.enums import AnalyticsEvent
@@ -26,6 +27,7 @@ from zenml.constants import (
     handle_int_env_var,
 )
 from zenml.enums import ExecutionStatus, StackComponentType, StoreType
+from zenml.exceptions import MaxConcurrentTasksError
 from zenml.logger import get_logger
 from zenml.models import (
     CodeReferenceRequest,
@@ -50,30 +52,92 @@ from zenml.zen_server.auth import AuthContext, generate_access_token
 from zenml.zen_server.template_execution.runner_entrypoint_configuration import (
     RunnerEntrypointConfiguration,
 )
-from zenml.zen_server.utils import server_config, workload_manager, zen_store
+from zenml.zen_server.utils import (
+    run_template_executor,
+    server_config,
+    workload_manager,
+    zen_store,
+)
 
 logger = get_logger(__name__)
 
 RUNNER_IMAGE_REPOSITORY = "zenml-runner"
 
 
+class BoundedThreadPoolExecutor:
+    """Thread pool executor which only allows a maximum number of concurrent tasks."""
+
+    def __init__(self, max_workers: int, **kwargs: Any) -> None:
+        """Initialize the executor.
+
+        Args:
+            max_workers: The maximum number of workers.
+            **kwargs: Arguments to pass to the thread pool executor.
+        """
+        self._executor = ThreadPoolExecutor(max_workers=max_workers, **kwargs)
+        self._semaphore = threading.BoundedSemaphore(value=max_workers)
+
+    def submit(
+        self, fn: Callable[..., Any], *args: Any, **kwargs: Any
+    ) -> Future[Any]:
+        """Submit a task to the executor.
+
+        Args:
+            fn: The function to execute.
+            *args: The arguments to pass to the function.
+            **kwargs: The keyword arguments to pass to the function.
+
+        Raises:
+            Exception: If the task submission fails.
+            MaxConcurrentTasksError: If the maximum number of concurrent tasks
+                is reached.
+
+        Returns:
+            The future of the task.
+        """
+        if not self._semaphore.acquire(blocking=False):
+            raise MaxConcurrentTasksError(
+                "Maximum number of concurrent tasks reached."
+            )
+
+        try:
+            future = self._executor.submit(fn, *args, **kwargs)
+        except Exception:
+            self._semaphore.release()
+            raise
+        else:
+            future.add_done_callback(lambda _: self._semaphore.release())
+            return future
+
+    def shutdown(self, **kwargs: Any) -> None:
+        """Shutdown the executor.
+
+        Args:
+            **kwargs: Keyword arguments to pass to the shutdown method of the
+                executor.
+        """
+        self._executor.shutdown(**kwargs)
+
+
 def run_template(
     template: RunTemplateResponse,
     auth_context: AuthContext,
-    background_tasks: Optional[BackgroundTasks] = None,
     run_config: Optional[PipelineRunConfiguration] = None,
+    sync: bool = False,
 ) -> PipelineRunResponse:
     """Run a pipeline from a template.
 
     Args:
         template: The template to run.
         auth_context: Authentication context.
-        background_tasks: Background tasks.
         run_config: The run configuration.
+        sync: Whether to run the template synchronously.
 
     Raises:
         ValueError: If the template can not be run.
         RuntimeError: If the server URL is not set in the server configuration.
+        MaxConcurrentTasksError: If the maximum number of concurrent run
+            template tasks is reached.
 
     Returns:
         ID of the new pipeline run.
@@ -241,13 +305,18 @@ def run_template(
                 )
                 raise
 
-    if background_tasks:
-        background_tasks.add_task(_task_with_analytics_and_error_handling)
-    else:
-        # Run synchronously if no background tasks were passed. This is probably
-        # when coming from a trigger which itself is already running in the
-        # background
+    if sync:
         _task_with_analytics_and_error_handling()
+    else:
+        try:
+            run_template_executor().submit(
+                _task_with_analytics_and_error_handling
+            )
+        except MaxConcurrentTasksError:
+            zen_store().delete_run(run_id=placeholder_run.id)
+            raise MaxConcurrentTasksError(
+                "Maximum number of concurrent run template tasks reached."
+            ) from None
 
     return placeholder_run
 

zenml/zen_server/utils.py

@@ -60,6 +60,9 @@ from zenml.zen_stores.sql_zen_store import SqlZenStore
 if TYPE_CHECKING:
     from fastapi import Request
 
+    from zenml.zen_server.template_execution.utils import (
+        BoundedThreadPoolExecutor,
+    )
 
 logger = get_logger(__name__)
 
@@ -67,6 +70,7 @@ _zen_store: Optional["SqlZenStore"] = None
 _rbac: Optional[RBACInterface] = None
 _feature_gate: Optional[FeatureGateInterface] = None
 _workload_manager: Optional[WorkloadManagerInterface] = None
+_run_template_executor: Optional["BoundedThreadPoolExecutor"] = None
 _plugin_flavor_registry: Optional[PluginFlavorRegistry] = None
 _memcache: Optional[MemoryCache] = None
 
@@ -196,6 +200,35 @@ def initialize_workload_manager() -> None:
             _workload_manager = workload_manager_class()
 
 
+def run_template_executor() -> "BoundedThreadPoolExecutor":
+    """Return the initialized run template executor.
+
+    Raises:
+        RuntimeError: If the run template executor is not initialized.
+
+    Returns:
+        The run template executor.
+    """
+    global _run_template_executor
+    if _run_template_executor is None:
+        raise RuntimeError("Run template executor not initialized")
+
+    return _run_template_executor
+
+
+def initialize_run_template_executor() -> None:
+    """Initialize the run template executor."""
+    global _run_template_executor
+    from zenml.zen_server.template_execution.utils import (
+        BoundedThreadPoolExecutor,
+    )
+
+    _run_template_executor = BoundedThreadPoolExecutor(
+        max_workers=server_config().max_concurrent_template_runs,
+        thread_name_prefix="zenml-run-template-executor",
+    )
+
+
 def initialize_plugins() -> None:
     """Initialize the event plugins registry."""
     plugin_flavor_registry().initialize_plugins()

zenml/zen_server/zen_server_api.py

@@ -102,9 +102,11 @@ from zenml.zen_server.utils import (
     initialize_memcache,
     initialize_plugins,
     initialize_rbac,
+    initialize_run_template_executor,
     initialize_workload_manager,
     initialize_zen_store,
     is_user_request,
+    run_template_executor,
     server_config,
     zen_store,
 )
@@ -390,6 +392,7 @@ def initialize() -> None:
     initialize_rbac()
     initialize_feature_gate()
     initialize_workload_manager()
+    initialize_run_template_executor()
     initialize_plugins()
     initialize_secure_headers()
     initialize_memcache(cfg.memcache_max_capacity, cfg.memcache_default_expiry)
@@ -399,6 +402,12 @@ def initialize() -> None:
         send_pro_workspace_status_update()
 
 
+@app.on_event("shutdown")
+def shutdown() -> None:
+    """Shutdown the ZenML server."""
+    run_template_executor().shutdown(wait=True)
+
+
 DASHBOARD_REDIRECT_URL = None
 if (
     server_config().dashboard_url

{zenml_nightly-0.82.0.dev20250506.dist-info → zenml_nightly-0.82.0.dev20250508.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.82.0.dev20250506
+Version: 0.82.0.dev20250508
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.82.0.dev20250506.dist-info → zenml_nightly-0.82.0.dev20250508.dist-info}/RECORD

@@ -1,10 +1,10 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=ppUfeGUhs-NasvRcl6HFrQq-3Bkc5Mvd-jli7V03Ufs,19
+zenml/VERSION,sha256=6ZapUMtAIHQPUp8zaMIANfD4zKfro3i2gKaHktdx2gk,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
 zenml/actions/pipeline_run/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-zenml/actions/pipeline_run/pipeline_run_action.py,sha256=eJx_V-DfWjP96DGZ_wP21jcKnbAbiffO5OTo68NNiTA,7438
+zenml/actions/pipeline_run/pipeline_run_action.py,sha256=0V0r-PcxYkmnjZ3X2efjs2ttUo4XIgTzjmoq1ei56Ew,7461
 zenml/alerter/__init__.py,sha256=2t0W_95iKIV_0iKOPLrViyynS5Cg-hswpcNw7lOCbVU,1047
 zenml/alerter/base_alerter.py,sha256=7g7iK0wKDNDYrugvO3RvbvHT0vzRyTtQfJvWwV_pl3k,3015
 zenml/analytics/__init__.py,sha256=jXTcNiyH1ZyUCR6NkoBmxQQnDXa2VGfkajpvfyRF6_Q,3064
@@ -77,7 +77,7 @@ zenml/config/retry_config.py,sha256=4UH1xqw0G6fSEbXSNKfmiFEkwadxQef9BGMe3JBm6NI,
 zenml/config/schedule.py,sha256=qtMWa-mEo7jIKvDzQUstMwe57gdbvyWAQ7ggsoddbCA,5349
 zenml/config/secret_reference_mixin.py,sha256=YvY68MTd1gE23IVprf0BLkNn62hoxcvb5nqGgc8jMkU,5871
 zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wzrx-zHTU,2818
-zenml/config/server_config.py,sha256=xLBOs_fWUM_Imk2IDzLPC1aVUSpwYYRobYAYC9hd2qE,31558
+zenml/config/server_config.py,sha256=or-LKYFl-9yE3MEq_yPa5aCCu3d5_f0tL9MqPtb_3-c,31852
 zenml/config/settings_resolver.py,sha256=PR9BRm_x1dy7nVKa9UqpeFdck8IEATSW6aWT8FKd-DI,4278
 zenml/config/source.py,sha256=RzUw8lin8QztUjz-AdoCzVM5Om_cSSPuroaPx-qAO4w,8226
 zenml/config/step_configurations.py,sha256=mngjobhHRj88f3klMdz6iw2mOj9wzYUPIV8Rp_2hV3g,10433
@@ -85,7 +85,7 @@ zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I
 zenml/config/store_config.py,sha256=Cla5p5dTB6nNlo8_OZDs9hod5hspi64vxwtZj882XgU,3559
 zenml/config/strict_base_model.py,sha256=t_ULrtJF2eW7TgyYBRobl1fscwwIZXATYky8ER97ev4,860
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
-zenml/constants.py,sha256=ZVwWVxE2h54cNbJDyujJmhAc8w3J7I9MkWHOfP4HN9A,16431
+zenml/constants.py,sha256=VI_-MR69Qxx3kEgxyRWL4Tf3wNAQyPYuJwsp6Yk59_Y,16485
 zenml/container_registries/__init__.py,sha256=ZSPbBIOnzhg88kQSpYgKe_POLuru14m629665-kAVAA,2200
 zenml/container_registries/azure_container_registry.py,sha256=t1sfDa94Vzbyqtb1iPFNutJ2EXV5_p9CUNITasoiQ70,2667
 zenml/container_registries/base_container_registry.py,sha256=-9RIkD6oXNPaU59R3PB_PtyCqsFoLPLSn5xYZmEmzbc,8915
@@ -110,7 +110,7 @@ zenml/event_sources/base_event.py,sha256=irrpiYi4fDYewzaXtb6_gPsqyFlYYLSap2BTGQz
 zenml/event_sources/base_event_source.py,sha256=04ramQhcPhGD9Mo9BcJZR-b9fJb8a4wzvlPVMAb9Dyk,25893
 zenml/event_sources/webhooks/__init__.py,sha256=VsHzSn1oKFaUs495ZqsKCfXiYxtaeGsnSGuvuqbGXCg,658
 zenml/event_sources/webhooks/base_webhook_event_source.py,sha256=2zADrL3cNpD-akZRdemUePUnTl8pPMPVX6eFQgWeUSo,7371
-zenml/exceptions.py,sha256=KPEz9kQGyC6VbgjNXFTUsVehvAUIe2ayOOy1j5QJdUI,6757
+zenml/exceptions.py,sha256=n69jnFh2KiQJsD5jU4mkSoxheCCu0L9g4FCf2aehzHQ,6883
 zenml/experiment_trackers/__init__.py,sha256=b5XlKoRtMR1WBQVEiItolkpsa0iJ1IqxTmmRatr4YDw,1119
 zenml/experiment_trackers/base_experiment_tracker.py,sha256=K92w7c0r45qLPIsA3YmwPflaIl_WTK8-_Hh_1MQiLkE,2218
 zenml/feature_stores/__init__.py,sha256=tSW7YnDa3NDnlkX3yA_CTdX7ocWB9gsfF-7X9sc6i8Y,1415
@@ -768,7 +768,7 @@ zenml/utils/daemon.py,sha256=GZ7Dx6GLHK04SR50wBxpKYmFhxPBfdLWxJiAWzJC6cM,11863
 zenml/utils/dashboard_utils.py,sha256=V4pas-zgMn3vOpXWZ-yoW11ZLZi__oFQw9jvhiXhK1A,8345
 zenml/utils/deprecation_utils.py,sha256=QcWkOnzIRDKPOfkr523n3l2MoY2wE0LIPfbx99t4Gmg,6343
 zenml/utils/dict_utils.py,sha256=i7KAaKrkaWA_cG5IvVfMnr0CwWlBJ7KAsGvP2wxjZI8,2667
-zenml/utils/docker_utils.py,sha256=QvkKnvIYSKAhW7mErXwSaQ432-q1LAsLjo2YWSXD8Bk,13889
+zenml/utils/docker_utils.py,sha256=_zKOVeTWuKJiQVtBP7y-6e4-kBfeWUAvXeM0ujXxDMM,14349
 zenml/utils/downloaded_repository_context.py,sha256=GyPOzC8M3SaEGnnbPgEgJhUW-m5hx9rLScnCiGw6_PY,1504
 zenml/utils/enum_utils.py,sha256=0fA0B9v9Wjutuqlu_owUoOit1utIw2UH5J6YHXSqhLU,1368
 zenml/utils/env_utils.py,sha256=2--2DDUt3gPOvCNVyobBtAciikQ0OEXs5WWp7NvYuKo,5276
@@ -1012,7 +1012,7 @@ zenml/zen_server/deploy/docker/docker_provider.py,sha256=18pNpxvP8xqbxS_KxvYTEIu
 zenml/zen_server/deploy/docker/docker_zen_server.py,sha256=313IXHgZLBPZx-lS9AF0CH0sSdoy3P1htS_d4ZXGuKw,7480
 zenml/zen_server/deploy/exceptions.py,sha256=tX0GNnLB_GMkeN7zGNlJRwtlrpZ5Slvyj_unVYVmGxk,1396
 zenml/zen_server/download_utils.py,sha256=g366TqytIWR4L4u9dd0jg-ouSWpDwdAh_-F2zryFzfo,4224
-zenml/zen_server/exceptions.py,sha256=NAXARDHLvfnVIQ87MHce0mfaWCmKhMKgwmw-_X49HxQ,9625
+zenml/zen_server/exceptions.py,sha256=hP8zJBofQljxPByFIa4tKpVjKdCAMJlKvk_5Opgusqg,9718
 zenml/zen_server/feature_gate/__init__.py,sha256=yabe4fBY6NSusn-QlKQDLOvXVLERNpcAQgigsyWQIbQ,612
 zenml/zen_server/feature_gate/endpoint_utils.py,sha256=o6sBVlqqlc9KokMaEsRTYeMra7f2a6kCt3FrB-oHhCw,2227
 zenml/zen_server/feature_gate/feature_gate_interface.py,sha256=XCgsqUN5yYCipodHCIDmCHRhyYMyt483Pp5mdpqzwHA,1647
@@ -1044,7 +1044,7 @@ zenml/zen_server/routers/pipelines_endpoints.py,sha256=R6OM-0Tc6ZNPzQmLs5MWTJIE4
 zenml/zen_server/routers/plugin_endpoints.py,sha256=vgIdA0kdYsf8PeF_pqDaqEYnIi2mlYALAv4uB9Nn7S0,3239
 zenml/zen_server/routers/projects_endpoints.py,sha256=iIgCg_0_iR5OddgULvsP4xxMOOmEGmtCuSKYZVRE9f4,8464
 zenml/zen_server/routers/run_metadata_endpoints.py,sha256=UTWQvweZTGuZnLrqhbIRYQ-z_TfOPY3Kn9p4YwU0tqw,3662
-zenml/zen_server/routers/run_templates_endpoints.py,sha256=EsiKxNVv5wvSkq_CqagxBfjeTh8Rnm37Vh1OynQrC-8,8338
+zenml/zen_server/routers/run_templates_endpoints.py,sha256=rEREmFmG38O_w6ktVJU5iu-dguGZpSC8RN9qH4WkEsA,8254
 zenml/zen_server/routers/runs_endpoints.py,sha256=YoTqWm-Mx97q8ZMYXgIY8gr2ICJpRlk6DxxHKN1utFU,12631
 zenml/zen_server/routers/schedule_endpoints.py,sha256=UXt6TIEX0PYhLoBgU-n5kIvWckX1BY_P5uUA_lK9J9A,5782
 zenml/zen_server/routers/secrets_endpoints.py,sha256=diYTC-jl6Hxd_BHVaNsPf2ZcWUwC99naWHsM1xRjR-g,9198
@@ -1064,10 +1064,10 @@ zenml/zen_server/routers/webhook_endpoints.py,sha256=KOJsuykv_TMjL3oEItpC4OWWP75
 zenml/zen_server/secure_headers.py,sha256=glh6QujnjyeoH1_FK-tAS-105G-qKS_34AqSzqJ6TRc,4182
 zenml/zen_server/template_execution/__init__.py,sha256=79knXLKfegsvVSVSWecpqrepq6iAavTUA4hKuiDk-WE,613
 zenml/zen_server/template_execution/runner_entrypoint_configuration.py,sha256=Y8aYJhqqs8Kv8I1q-dM1WemS5VBIfyoaaYH_YkzC7iY,1541
-zenml/zen_server/template_execution/utils.py,sha256=UGMFeyVB8Os2awUfXsvKvDLdMiT0uxRj7ErF-8RyhA0,17533
+zenml/zen_server/template_execution/utils.py,sha256=jx_FdkrXKI-LOG1DykedhpNPwoUm_iFDJF9r1UVaE_o,19680
 zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8ajuZE01DaHmdCDCZmsroDcTarvN-nE8jv6qQ,2590
-zenml/zen_server/utils.py,sha256=Jc2Q4UBaYG2ruHdsN9JmbOWfWU_eWD9wTBBEgcGAbqg,17439
-zenml/zen_server/zen_server_api.py,sha256=ALyv5096frXXRNySegEtsmkDbHLLqHd402bbQI7RnII,17941
+zenml/zen_server/utils.py,sha256=yOwPI0DxzDIxF7Ppnty_DuGMWYJ_ubHUNWfbTQ1_q9Q,18461
+zenml/zen_server/zen_server_api.py,sha256=1_YHOUmBulXuHWCxDLYtDCMxGze159UmU3f05643PSg,18182
 zenml/zen_stores/__init__.py,sha256=6LTgH6XwDeDxKqVJ1JTfGhmS8II1NLopPloINGmdyI0,691
 zenml/zen_stores/base_zen_store.py,sha256=AplsW2NR-G9_CU54XvNTQJo4W0KJ5TJV22cjKW4n2BY,16124
 zenml/zen_stores/migrations/README.md,sha256=x04jsb6EOP6PBEGMQlDELiqKEham2O-iztAs9AylMFc,4898
@@ -1318,8 +1318,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=biOoDb2_zYmpsN-J-FSlKICYdwM9KDIe-_KN_yDf_mA,441414
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.82.0.dev20250506.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.82.0.dev20250506.dist-info/METADATA,sha256=_sWp9WjyqeWPdnHJ3RUVzjNOID0vYkZhrN-yKVh7tuM,24315
-zenml_nightly-0.82.0.dev20250506.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.82.0.dev20250506.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.82.0.dev20250506.dist-info/RECORD,,
+zenml_nightly-0.82.0.dev20250508.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.82.0.dev20250508.dist-info/METADATA,sha256=HwiVOaLegimi0ZV8nGTInes1CvJnC-FMDRC-eu9bAgw,24315
+zenml_nightly-0.82.0.dev20250508.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.82.0.dev20250508.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.82.0.dev20250508.dist-info/RECORD,,