zenml-nightly 0.82.0.dev20250509__py3-none-any.whl → 0.82.0.dev20250511__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.82.0.dev20250509
+0.82.0.dev20250511

zenml/config/compiler.py

@@ -27,6 +27,7 @@ from typing import (
 
 from zenml import __version__
 from zenml.config.base_settings import BaseSettings, ConfigurationLevel
+from zenml.config.pipeline_configurations import PipelineConfiguration
 from zenml.config.pipeline_run_configuration import PipelineRunConfiguration
 from zenml.config.pipeline_spec import PipelineSpec
 from zenml.config.settings_resolver import SettingsResolver
@@ -43,7 +44,6 @@ from zenml.pipelines.run_utils import get_default_run_name
 from zenml.utils import pydantic_utils, settings_utils
 
 if TYPE_CHECKING:
-    from zenml.config.source import Source
     from zenml.pipelines.pipeline_definition import Pipeline
     from zenml.stack import Stack, StackComponent
     from zenml.steps.step_invocation import StepInvocation
@@ -112,21 +112,12 @@ class Compiler:
         with pipeline.__suppress_configure_warnings__():
             pipeline.configure(settings=pipeline_settings, merge=False)
 
-        settings_to_passdown = {
-            key: settings
-            for key, settings in pipeline_settings.items()
-            if ConfigurationLevel.STEP in settings.LEVEL
-        }
-
         steps = {
             invocation_id: self._compile_step_invocation(
                 invocation=invocation,
-                pipeline_settings=settings_to_passdown,
-                pipeline_extra=pipeline.configuration.extra,
                 stack=stack,
                 step_config=run_configuration.steps.get(invocation_id),
-                pipeline_failure_hook_source=pipeline.configuration.failure_hook_source,
-                pipeline_success_hook_source=pipeline.configuration.success_hook_source,
+                pipeline_configuration=pipeline.configuration,
             )
             for invocation_id, invocation in self._get_sorted_invocations(
                 pipeline=pipeline
@@ -447,24 +438,17 @@ class Compiler:
     def _compile_step_invocation(
         self,
         invocation: "StepInvocation",
-        pipeline_settings: Dict[str, "BaseSettings"],
-        pipeline_extra: Dict[str, Any],
         stack: "Stack",
         step_config: Optional["StepConfigurationUpdate"],
-        pipeline_failure_hook_source: Optional["Source"] = None,
-        pipeline_success_hook_source: Optional["Source"] = None,
+        pipeline_configuration: "PipelineConfiguration",
     ) -> Step:
         """Compiles a ZenML step.
 
         Args:
             invocation: The step invocation to compile.
-            pipeline_settings: settings configured on the
-                pipeline of the step.
-            pipeline_extra: Extra values configured on the pipeline of the step.
             stack: The stack on which the pipeline will be run.
             step_config: Run configuration for the step.
-            pipeline_failure_hook_source: Source for the failure hook.
-            pipeline_success_hook_source: Source for the success hook.
+            pipeline_configuration: Configuration for the pipeline.
 
         Returns:
             The compiled step.
@@ -488,32 +472,27 @@ class Compiler:
             configuration_level=ConfigurationLevel.STEP,
             stack=stack,
         )
-        step_extra = step.configuration.extra
-        step_on_failure_hook_source = step.configuration.failure_hook_source
-        step_on_success_hook_source = step.configuration.success_hook_source
-
-        step.configure(
-            settings=pipeline_settings,
-            extra=pipeline_extra,
-            on_failure=pipeline_failure_hook_source,
-            on_success=pipeline_success_hook_source,
-            merge=False,
-        )
         step.configure(
             settings=step_settings,
-            extra=step_extra,
-            on_failure=step_on_failure_hook_source,
-            on_success=step_on_success_hook_source,
-            merge=True,
+            merge=False,
         )
 
         parameters_to_ignore = (
             set(step_config.parameters) if step_config else set()
         )
-        complete_step_configuration = invocation.finalize(
+        step_configuration_overrides = invocation.finalize(
             parameters_to_ignore=parameters_to_ignore
         )
-        return Step(spec=step_spec, config=complete_step_configuration)
+        full_step_config = (
+            step_configuration_overrides.apply_pipeline_configuration(
+                pipeline_configuration=pipeline_configuration
+            )
+        )
+        return Step(
+            spec=step_spec,
+            config=full_step_config,
+            step_config_overrides=step_configuration_overrides,
+        )
 
     def _get_sorted_invocations(
         self,

zenml/config/pipeline_configurations.py

@@ -52,23 +52,30 @@ class PipelineConfigurationUpdate(StrictBaseModel):
     retry: Optional[StepRetryConfig] = None
     substitutions: Dict[str, str] = {}
 
-    def _get_full_substitutions(
-        self, start_time: Optional[datetime]
+    def finalize_substitutions(
+        self, start_time: Optional[datetime] = None, inplace: bool = False
     ) -> Dict[str, str]:
         """Returns the full substitutions dict.
 
         Args:
             start_time: Start time of the pipeline run.
+            inplace: Whether to update the substitutions in place.
 
         Returns:
             The full substitutions dict including date and time.
         """
         if start_time is None:
             start_time = utc_now()
-        ret = self.substitutions.copy()
-        ret.setdefault("date", start_time.strftime("%Y_%m_%d"))
-        ret.setdefault("time", start_time.strftime("%H_%M_%S_%f"))
-        return ret
+
+        if inplace:
+            dict_ = self.substitutions
+        else:
+            dict_ = self.substitutions.copy()
+
+        dict_.setdefault("date", start_time.strftime("%Y_%m_%d"))
+        dict_.setdefault("time", start_time.strftime("%H_%M_%S_%f"))
+
+        return dict_
 
 
 class PipelineConfiguration(PipelineConfigurationUpdate):

zenml/config/step_configurations.py

@@ -13,7 +13,6 @@
 #  permissions and limitations under the License.
 """Pipeline configuration classes."""
 
-from datetime import datetime
 from typing import (
     TYPE_CHECKING,
     Any,
@@ -46,7 +45,7 @@ from zenml.logger import get_logger
 from zenml.model.lazy_load import ModelVersionDataLazyLoader
 from zenml.model.model import Model
 from zenml.utils import deprecation_utils
-from zenml.utils.pydantic_utils import before_validator_handler
+from zenml.utils.pydantic_utils import before_validator_handler, update_model
 
 if TYPE_CHECKING:
     from zenml.config import DockerSettings, ResourceSettings
@@ -241,23 +240,43 @@ class StepConfiguration(PartialStepConfiguration):
             model_or_dict = model_or_dict.model_dump()
         return DockerSettings.model_validate(model_or_dict)
 
-    def _get_full_substitutions(
-        self,
-        pipeline_config: "PipelineConfiguration",
-        start_time: Optional[datetime],
-    ) -> Dict[str, str]:
-        """Get the full set of substitutions for this step configuration.
+    def apply_pipeline_configuration(
+        self, pipeline_configuration: "PipelineConfiguration"
+    ) -> "StepConfiguration":
+        """Apply the pipeline configuration to this step configuration.
 
         Args:
-            pipeline_config: The pipeline configuration.
-            start_time: The start time of the pipeline run.
+            pipeline_configuration: The pipeline configuration to apply.
 
         Returns:
-            The full set of substitutions for this step configuration.
+            The updated step configuration.
         """
-        ret = pipeline_config._get_full_substitutions(start_time)
-        ret.update(self.substitutions)
-        return ret
+        pipeline_values = pipeline_configuration.model_dump(
+            include={
+                "settings",
+                "extra",
+                "failure_hook_source",
+                "success_hook_source",
+                "substitutions",
+            },
+            exclude_none=True,
+        )
+        if pipeline_values:
+            original_values = self.model_dump(
+                include={
+                    "settings",
+                    "extra",
+                    "failure_hook_source",
+                    "success_hook_source",
+                    "substitutions",
+                },
+                exclude_none=True,
+            )
+
+            updated_config = self.model_copy(update=pipeline_values, deep=True)
+            return update_model(updated_config, original_values)
+        else:
+            return self.model_copy(deep=True)
 
 
 class InputSpec(StrictBaseModel):
@@ -311,3 +330,59 @@ class Step(StrictBaseModel):
 
     spec: StepSpec
     config: StepConfiguration
+    step_config_overrides: StepConfiguration
+
+    @model_validator(mode="before")
+    @classmethod
+    @before_validator_handler
+    def _add_step_config_overrides_if_missing(cls, data: Any) -> Any:
+        """Add step config overrides if missing.
+
+        This is to ensure backwards compatibility with data stored in the DB
+        before the `step_config_overrides` field was added. In that case, only
+        the `config` field, which contains the merged pipeline and step configs,
+        existed. We have no way to figure out which of those values were defined
+        on the step vs the pipeline level, so we just use the entire `config`
+        object as the `step_config_overrides`.
+
+        Args:
+            data: The values dict used to instantiate the model.
+
+        Returns:
+            The values dict with the step config overrides added if missing.
+        """
+        if "step_config_overrides" not in data:
+            data["step_config_overrides"] = data["config"]
+
+        return data
+
+    @classmethod
+    def from_dict(
+        cls,
+        data: Dict[str, Any],
+        pipeline_configuration: "PipelineConfiguration",
+    ) -> "Step":
+        """Create a step from a dictionary.
+
+        This method can create a step from data stored without the merged
+        `config` attribute, by merging the `step_config_overrides` with the
+        pipeline configuration.
+
+        Args:
+            data: The dictionary to create the `Step` object from.
+            pipeline_configuration: The pipeline configuration to apply to the
+                step configuration.
+
+        Returns:
+            The instantiated object.
+        """
+        if "config" not in data:
+            config = StepConfiguration.model_validate(
+                data["step_config_overrides"]
+            )
+            config = config.apply_pipeline_configuration(
+                pipeline_configuration
+            )
+            data["config"] = config
+
+        return cls.model_validate(data)

zenml/constants.py

@@ -199,6 +199,7 @@ ENV_ZENML_RUN_SINGLE_STEPS_WITHOUT_STACK = (
 )
 ENV_ZENML_PREVENT_CLIENT_SIDE_CACHING = "ZENML_PREVENT_CLIENT_SIDE_CACHING"
 ENV_ZENML_DISABLE_CREDENTIALS_DISK_CACHING = "DISABLE_CREDENTIALS_DISK_CACHING"
+ENV_ZENML_RUNNER_PARENT_IMAGE = "ZENML_RUNNER_PARENT_IMAGE"
 ENV_ZENML_RUNNER_IMAGE_DISABLE_UV = "ZENML_RUNNER_IMAGE_DISABLE_UV"
 ENV_ZENML_RUNNER_POD_TIMEOUT = "ZENML_RUNNER_POD_TIMEOUT"
 ENV_ZENML_WORKLOAD_TOKEN_EXPIRATION_LEEWAY = (

zenml/models/v2/base/scoped.py

@@ -318,6 +318,8 @@ class ProjectScopedResponse(
     Used as a base class for all domain models that are project-scoped.
     """
 
+    project_id: UUID = Field(title="The project id.")
+
     # Analytics
     def get_analytics_metadata(self) -> Dict[str, Any]:
         """Fetches the analytics metadata for project scoped models.

zenml/models/v2/core/artifact_version.py

@@ -710,6 +710,7 @@ class LazyArtifactVersionResponse(ArtifactVersionResponse):
     """
 
     id: Optional[UUID] = None  # type: ignore[assignment]
+    project_id: Optional[UUID] = None  # type: ignore[assignment]
     lazy_load_name: Optional[str] = None
     lazy_load_version: Optional[str] = None
     lazy_load_model_name: str

zenml/orchestrators/step_launcher.py

@@ -308,8 +308,8 @@ class StepLauncher:
         start_time = utc_now()
         run_name = string_utils.format_name_template(
             name_template=self._deployment.run_name_template,
-            substitutions=self._deployment.pipeline_configuration._get_full_substitutions(
-                start_time
+            substitutions=self._deployment.pipeline_configuration.finalize_substitutions(
+                start_time=start_time,
             ),
         )
 

zenml/pipelines/run_utils.py

@@ -72,8 +72,8 @@ def create_placeholder_run(
     run_request = PipelineRunRequest(
         name=string_utils.format_name_template(
             name_template=deployment.run_name_template,
-            substitutions=deployment.pipeline_configuration._get_full_substitutions(
-                start_time
+            substitutions=deployment.pipeline_configuration.finalize_substitutions(
+                start_time=start_time,
             ),
         ),
         # We set the start time on the placeholder run already to
@@ -113,6 +113,7 @@ def get_placeholder_run(
         size=1,
         deployment_id=deployment_id,
         status=ExecutionStatus.INITIALIZING,
+        hydrate=True,
     )
     if len(runs.items) == 0:
         return None

zenml/stack/stack_component.py

@@ -529,10 +529,13 @@ class StackComponent:
             else container.pipeline_configuration.settings
         )
 
+        # Use the current config as a base
+        settings_dict = self.config.model_dump()
+
         if key in all_settings:
-            return self.settings_class.model_validate(dict(all_settings[key]))
-        else:
-            return self.settings_class()
+            settings_dict.update(dict(all_settings[key]))
+
+        return self.settings_class.model_validate(settings_dict)
 
     def connector_has_expired(self) -> bool:
         """Checks whether the connector linked to this stack component has expired.

zenml/zen_server/rbac/utils.py

@@ -373,8 +373,7 @@ def get_resource_for_model(model: AnyModel) -> Optional[Resource]:
 
     project_id: Optional[UUID] = None
     if isinstance(model, ProjectScopedResponse):
-        # A project scoped response is always scoped to a specific project
-        project_id = model.project.id
+        project_id = model.project_id
     elif isinstance(model, ProjectScopedRequest):
         # A project scoped request is always scoped to a specific project
         project_id = model.project

zenml/zen_server/routers/pipeline_deployments_endpoints.py

@@ -13,18 +13,16 @@
 #  permissions and limitations under the License.
 """Endpoint definitions for deployments."""
 
-from typing import Optional, Union
+from typing import Any, Optional, Union
 from uuid import UUID
 
-from fastapi import APIRouter, Depends, Security
+from fastapi import APIRouter, Depends, Request, Security
 
 from zenml.constants import API, PIPELINE_DEPLOYMENTS, VERSION_1
 from zenml.logging.step_logging import fetch_logs
 from zenml.models import (
-    Page,
     PipelineDeploymentFilter,
     PipelineDeploymentRequest,
-    PipelineDeploymentResponse,
     PipelineRunFilter,
 )
 from zenml.zen_server.auth import AuthContext, authorize
@@ -45,6 +43,37 @@ from zenml.zen_server.utils import (
     zen_store,
 )
 
+
+# TODO: Remove this as soon as there is only a low number of users running
+# versions < 0.82.0. Also change back the return type to
+# `PipelineDeploymentResponse` once we have removed the `exclude` logic.
+def _should_remove_step_config_overrides(
+    request: Request,
+) -> bool:
+    """Check if the step config overrides should be removed from the response.
+
+    Args:
+        request: The request object.
+
+    Returns:
+        If the step config overrides should be removed from the response.
+    """
+    from packaging import version
+
+    user_agent = request.headers.get("User-Agent", "")
+
+    if not user_agent.startswith("zenml/"):
+        # This request is not coming from a ZenML client
+        return False
+
+    client_version = version.parse(user_agent.removeprefix("zenml/"))
+
+    # Versions before 0.82.0 did have `extra="forbid"` in the pydantic model
+    # that stores the step configurations. This means it would crash if we
+    # included the `step_config_overrides` in the response.
+    return client_version < version.parse("0.82.0")
+
+
 router = APIRouter(
     prefix=API + VERSION_1 + PIPELINE_DEPLOYMENTS,
     tags=["deployments"],
@@ -66,13 +95,15 @@ router = APIRouter(
 )
 @handle_exceptions
 def create_deployment(
+    request: Request,
     deployment: PipelineDeploymentRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
-) -> PipelineDeploymentResponse:
+) -> Any:
     """Creates a deployment.
 
     Args:
+        request: The request object.
         deployment: Deployment to create.
         project_name_or_id: Optional name or ID of the project.
 
@@ -83,11 +114,21 @@ def create_deployment(
         project = zen_store().get_project(project_name_or_id)
         deployment.project = project.id
 
-    return verify_permissions_and_create_entity(
+    deployment_response = verify_permissions_and_create_entity(
         request_model=deployment,
         create_method=zen_store().create_deployment,
     )
 
+    exclude = None
+    if _should_remove_step_config_overrides(request):
+        exclude = {
+            "metadata": {
+                "step_configurations": {"__all__": {"step_config_overrides"}}
+            }
+        }
+
+    return deployment_response.model_dump(mode="json", exclude=exclude)
+
 
 @router.get(
     "",
@@ -103,16 +144,18 @@ def create_deployment(
 )
 @handle_exceptions
 def list_deployments(
+    request: Request,
     deployment_filter_model: PipelineDeploymentFilter = Depends(
         make_dependable(PipelineDeploymentFilter)
     ),
     project_name_or_id: Optional[Union[str, UUID]] = None,
     hydrate: bool = False,
     _: AuthContext = Security(authorize),
-) -> Page[PipelineDeploymentResponse]:
+) -> Any:
     """Gets a list of deployments.
 
     Args:
+        request: The request object.
         deployment_filter_model: Filter model used for pagination, sorting,
             filtering.
         project_name_or_id: Optional name or ID of the project to filter by.
@@ -125,13 +168,29 @@ def list_deployments(
     if project_name_or_id:
         deployment_filter_model.project = project_name_or_id
 
-    return verify_permissions_and_list_entities(
+    page = verify_permissions_and_list_entities(
         filter_model=deployment_filter_model,
         resource_type=ResourceType.PIPELINE_DEPLOYMENT,
         list_method=zen_store().list_deployments,
         hydrate=hydrate,
     )
 
+    exclude = None
+    if _should_remove_step_config_overrides(request):
+        exclude = {
+            "items": {
+                "__all__": {
+                    "metadata": {
+                        "step_configurations": {
+                            "__all__": {"step_config_overrides"}
+                        }
+                    }
+                }
+            }
+        }
+
+    return page.model_dump(mode="json", exclude=exclude)
+
 
 @router.get(
     "/{deployment_id}",
@@ -139,13 +198,15 @@ def list_deployments(
 )
 @handle_exceptions
 def get_deployment(
+    request: Request,
     deployment_id: UUID,
     hydrate: bool = True,
     _: AuthContext = Security(authorize),
-) -> PipelineDeploymentResponse:
+) -> Any:
     """Gets a specific deployment using its unique id.
 
     Args:
+        request: The request object.
         deployment_id: ID of the deployment to get.
         hydrate: Flag deciding whether to hydrate the output model(s)
             by including metadata fields in the response.
@@ -153,12 +214,22 @@ def get_deployment(
     Returns:
         A specific deployment object.
     """
-    return verify_permissions_and_get_entity(
+    deployment = verify_permissions_and_get_entity(
         id=deployment_id,
         get_method=zen_store().get_deployment,
         hydrate=hydrate,
     )
 
+    exclude = None
+    if _should_remove_step_config_overrides(request):
+        exclude = {
+            "metadata": {
+                "step_configurations": {"__all__": {"step_config_overrides"}}
+            }
+        }
+
+    return deployment.model_dump(mode="json", exclude=exclude)
+
 
 @router.delete(
     "/{deployment_id}",

zenml/zen_server/template_execution/utils.py

@@ -1,7 +1,7 @@
 """Utility functions to run a pipeline from the server."""
 
-import copy
 import hashlib
+import os
 import sys
 import threading
 from concurrent.futures import Future, ThreadPoolExecutor
@@ -17,11 +17,12 @@ from zenml.config.pipeline_configurations import PipelineConfiguration
 from zenml.config.pipeline_run_configuration import (
     PipelineRunConfiguration,
 )
-from zenml.config.step_configurations import Step, StepConfiguration
+from zenml.config.step_configurations import Step, StepConfigurationUpdate
 from zenml.constants import (
     ENV_ZENML_ACTIVE_PROJECT_ID,
     ENV_ZENML_ACTIVE_STACK_ID,
     ENV_ZENML_RUNNER_IMAGE_DISABLE_UV,
+    ENV_ZENML_RUNNER_PARENT_IMAGE,
     ENV_ZENML_RUNNER_POD_TIMEOUT,
     handle_bool_env_var,
     handle_int_env_var,
@@ -47,7 +48,7 @@ from zenml.pipelines.run_utils import (
     validate_stack_is_runnable_from_server,
 )
 from zenml.stack.flavor import Flavor
-from zenml.utils import dict_utils, requirements_utils, settings_utils
+from zenml.utils import pydantic_utils, requirements_utils, settings_utils
 from zenml.zen_server.auth import AuthContext, generate_access_token
 from zenml.zen_server.template_execution.runner_entrypoint_configuration import (
     RunnerEntrypointConfiguration,
@@ -172,7 +173,6 @@ def run_template(
     deployment_request = deployment_request_from_template(
         template=template,
         config=run_config or PipelineRunConfiguration(),
-        user_id=auth_context.user.id,
     )
 
     ensure_async_orchestrator(deployment=deployment_request, stack=stack)
@@ -386,7 +386,10 @@ def generate_dockerfile(
     Returns:
         The Dockerfile.
     """
-    parent_image = f"zenmldocker/zenml:{zenml_version}-py{python_version}"
+    parent_image = os.environ.get(
+        ENV_ZENML_RUNNER_PARENT_IMAGE,
+        f"zenmldocker/zenml:{zenml_version}-py{python_version}",
+    )
 
     lines = [f"FROM {parent_image}"]
     if apt_packages:
@@ -420,14 +423,12 @@ def generate_dockerfile(
 def deployment_request_from_template(
     template: RunTemplateResponse,
     config: PipelineRunConfiguration,
-    user_id: UUID,
 ) -> "PipelineDeploymentRequest":
     """Generate a deployment request from a template.
 
     Args:
         template: The template from which to create the deployment request.
         config: The run configuration.
-        user_id: ID of the user that is trying to run the template.
 
     Raises:
         ValueError: If there are missing/extra step parameters in the run
@@ -438,49 +439,37 @@ def deployment_request_from_template(
     """
     deployment = template.source_deployment
     assert deployment
-    pipeline_configuration = PipelineConfiguration(
-        **config.model_dump(
-            include=set(PipelineConfiguration.model_fields),
-            exclude={"name", "parameters"},
-        ),
-        name=deployment.pipeline_configuration.name,
-        parameters=deployment.pipeline_configuration.parameters,
-    )
 
-    step_config_dict_base = pipeline_configuration.model_dump(
-        exclude={"name", "parameters", "tags", "enable_pipeline_logs"}
+    pipeline_update = config.model_dump(
+        include=set(PipelineConfiguration.model_fields),
+        exclude={"name", "parameters"},
+        exclude_unset=True,
+        exclude_none=True,
+    )
+    pipeline_configuration = pydantic_utils.update_model(
+        deployment.pipeline_configuration, pipeline_update
     )
+
     steps = {}
     for invocation_id, step in deployment.step_configurations.items():
-        step_config_dict = {
-            **copy.deepcopy(step_config_dict_base),
-            **step.config.model_dump(
-                # TODO: Maybe we need to make some of these configurable via
-                # yaml as well, e.g. the lazy loaders?
-                include={
-                    "name",
-                    "caching_parameters",
-                    "external_input_artifacts",
-                    "model_artifacts_or_metadata",
-                    "client_lazy_loaders",
-                    "substitutions",
-                    "outputs",
-                }
-            ),
-        }
-
-        required_parameters = set(step.config.parameters)
-        configured_parameters = set()
-
-        if update := config.steps.get(invocation_id):
-            update_dict = update.model_dump()
+        step_update = config.steps.get(
+            invocation_id, StepConfigurationUpdate()
+        ).model_dump(
             # Get rid of deprecated name to prevent overriding the step name
             # with `None`.
-            update_dict.pop("name", None)
-            configured_parameters = set(update.parameters)
-            step_config_dict = dict_utils.recursive_update(
-                step_config_dict, update=update_dict
-            )
+            exclude={"name"},
+            exclude_unset=True,
+            exclude_none=True,
+        )
+        step_config = pydantic_utils.update_model(
+            step.step_config_overrides, step_update
+        )
+        merged_step_config = step_config.apply_pipeline_configuration(
+            pipeline_configuration
+        )
+
+        required_parameters = set(step.config.parameters)
+        configured_parameters = set(step_config.parameters)
 
         unknown_parameters = configured_parameters - required_parameters
         if unknown_parameters:
@@ -496,8 +485,11 @@ def deployment_request_from_template(
                 f"parameters for step {invocation_id}: {missing_parameters}."
             )
 
-        step_config = StepConfiguration.model_validate(step_config_dict)
-        steps[invocation_id] = Step(spec=step.spec, config=step_config)
+        steps[invocation_id] = Step(
+            spec=step.spec,
+            config=merged_step_config,
+            step_config_overrides=step_config,
+        )
 
     code_reference_request = None
     if deployment.code_reference:

zenml/zen_stores/schemas/action_schemas.py

@@ -192,6 +192,7 @@ class ActionSchema(NamedSchema, table=True):
             )
         return ActionResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/artifact_schemas.py

@@ -203,6 +203,7 @@ class ArtifactSchema(NamedSchema, table=True):
 
         return ArtifactResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,
@@ -437,6 +438,7 @@ class ArtifactVersionSchema(BaseSchema, RunMetadataInterface, table=True):
 
         return ArtifactVersionResponse(
             id=self.id,
+            project_id=self.project_id,
             body=body,
             metadata=metadata,
             resources=resources,

zenml/zen_stores/schemas/code_repository_schemas.py

@@ -133,6 +133,7 @@ class CodeRepositorySchema(NamedSchema, table=True):
             )
         return CodeRepositoryResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             metadata=metadata,
             body=body,

zenml/zen_stores/schemas/event_source_schemas.py

@@ -165,6 +165,7 @@ class EventSourceSchema(NamedSchema, table=True):
             )
         return EventSourceResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/model_schemas.py

@@ -228,6 +228,7 @@ class ModelSchema(NamedSchema, table=True):
 
         return ModelResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,
@@ -502,6 +503,7 @@ class ModelVersionSchema(NamedSchema, RunMetadataInterface, table=True):
 
         return ModelVersionResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/pipeline_build_schemas.py

@@ -169,6 +169,7 @@ class PipelineBuildSchema(BaseSchema, table=True):
             )
         return PipelineBuildResponse(
             id=self.id,
+            project_id=self.project_id,
             body=body,
             metadata=metadata,
         )

zenml/zen_stores/schemas/pipeline_deployment_schemas.py

@@ -32,7 +32,6 @@ from zenml.models import (
     PipelineDeploymentResponseBody,
     PipelineDeploymentResponseMetadata,
 )
-from zenml.utils.json_utils import pydantic_encoder
 from zenml.zen_stores.schemas.base_schemas import BaseSchema
 from zenml.zen_stores.schemas.code_repository_schemas import (
     CodeReferenceSchema,
@@ -188,6 +187,14 @@ class PipelineDeploymentSchema(BaseSchema, table=True):
         Returns:
             The created `PipelineDeploymentSchema`.
         """
+        # Don't include the merged config in the step configurations, we
+        # reconstruct it in the `to_model` method using the pipeline
+        # configuration.
+        step_configurations = {
+            invocation_id: step.model_dump(mode="json", exclude={"config"})
+            for invocation_id, step in request.step_configurations.items()
+        }
+
         client_env = json.dumps(request.client_environment)
         if len(client_env) > TEXT_FIELD_MAX_LENGTH:
             logger.warning(
@@ -208,9 +215,8 @@ class PipelineDeploymentSchema(BaseSchema, table=True):
             run_name_template=request.run_name_template,
             pipeline_configuration=request.pipeline_configuration.model_dump_json(),
             step_configurations=json.dumps(
-                request.step_configurations,
+                step_configurations,
                 sort_keys=False,
-                default=pydantic_encoder,
             ),
             client_environment=client_env,
             client_version=request.client_version,
@@ -254,8 +260,10 @@ class PipelineDeploymentSchema(BaseSchema, table=True):
                 self.pipeline_configuration
             )
             step_configurations = json.loads(self.step_configurations)
-            for s, c in step_configurations.items():
-                step_configurations[s] = Step.model_validate(c)
+            for invocation_id, step in step_configurations.items():
+                step_configurations[invocation_id] = Step.from_dict(
+                    step, pipeline_configuration
+                )
 
             client_environment = json.loads(self.client_environment)
             if not include_python_packages:
@@ -287,6 +295,7 @@ class PipelineDeploymentSchema(BaseSchema, table=True):
             )
         return PipelineDeploymentResponse(
             id=self.id,
+            project_id=self.project_id,
             body=body,
             metadata=metadata,
         )

zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -317,10 +317,6 @@ class PipelineRunSchema(NamedSchema, RunMetadataInterface, table=True):
             )
 
             config = deployment.pipeline_configuration
-            new_substitutions = config._get_full_substitutions(self.start_time)
-            config = config.model_copy(
-                update={"substitutions": new_substitutions}
-            )
             client_environment = deployment.client_environment
 
             stack = deployment.stack
@@ -352,6 +348,8 @@ class PipelineRunSchema(NamedSchema, RunMetadataInterface, table=True):
                 "pipeline_configuration."
             )
 
+        config.finalize_substitutions(start_time=self.start_time, inplace=True)
+
         body = PipelineRunResponseBody(
             user=self.user.to_model() if self.user else None,
             status=ExecutionStatus(self.status),
@@ -437,6 +435,7 @@ class PipelineRunSchema(NamedSchema, RunMetadataInterface, table=True):
 
         return PipelineRunResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/pipeline_schemas.py

@@ -196,6 +196,7 @@ class PipelineSchema(NamedSchema, table=True):
 
         return PipelineResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/run_template_schemas.py

@@ -299,6 +299,7 @@ class RunTemplateSchema(NamedSchema, table=True):
 
         return RunTemplateResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/schedule_schema.py

@@ -211,6 +211,7 @@ class ScheduleSchema(NamedSchema, RunMetadataInterface, table=True):
 
         return ScheduleResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/service_schemas.py

@@ -171,6 +171,7 @@ class ServiceSchema(NamedSchema, table=True):
             )
         return ServiceResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/step_run_schemas.py

@@ -232,9 +232,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
             The created StepRunResponse.
 
         Raises:
-            ValueError: In case the step run configuration can not be loaded.
-            RuntimeError: If the step run schema does not have a deployment_id
-                or a step_configuration.
+            ValueError: In case the step run configuration is missing.
         """
         input_artifacts = {
             artifact.name: StepRunInputResponse(
@@ -252,49 +250,38 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
                 artifact.artifact_version.to_model()
             )
 
-        full_step_config = None
+        step = None
         if self.deployment is not None:
-            step_configuration = json.loads(
+            step_configurations = json.loads(
                 self.deployment.step_configurations
             )
-            if self.name in step_configuration:
-                full_step_config = Step.model_validate(
-                    step_configuration[self.name]
-                )
-                new_substitutions = (
-                    full_step_config.config._get_full_substitutions(
-                        PipelineConfiguration.model_validate_json(
-                            self.deployment.pipeline_configuration
-                        ),
-                        self.pipeline_run.start_time,
+            if self.name in step_configurations:
+                pipeline_configuration = (
+                    PipelineConfiguration.model_validate_json(
+                        self.deployment.pipeline_configuration
                     )
                 )
-                full_step_config = full_step_config.model_copy(
-                    update={
-                        "config": full_step_config.config.model_copy(
-                            update={"substitutions": new_substitutions}
-                        )
-                    }
-                )
-            elif not self.step_configuration:
-                raise ValueError(
-                    f"Unable to load the configuration for step `{self.name}` from the"
-                    f"database. To solve this please delete the pipeline run that this"
-                    f"step run belongs to. Pipeline Run ID: `{self.pipeline_run_id}`."
+                pipeline_configuration.finalize_substitutions(
+                    start_time=self.pipeline_run.start_time,
+                    inplace=True,
                 )
-
-        # the step configuration moved into the deployment - the following case is to ensure
-        # backwards compatibility
-        if full_step_config is None:
-            if self.step_configuration:
-                full_step_config = Step.model_validate_json(
-                    self.step_configuration
-                )
-            else:
-                raise RuntimeError(
-                    "Step run model creation has failed. Each step run entry "
-                    "should either have a deployment_id or step_configuration."
+                step = Step.from_dict(
+                    step_configurations[self.name],
+                    pipeline_configuration=pipeline_configuration,
                 )
+        if not step and self.step_configuration:
+            # In this legacy case, we're guaranteed to have the merged
+            # config stored in the DB, which means we can instantiate the
+            # `Step` object directly without passing the pipeline
+            # configuration.
+            step = Step.model_validate_json(self.step_configuration)
+        elif not step:
+            raise ValueError(
+                f"Unable to load the configuration for step `{self.name}` from "
+                "the database. To solve this please delete the pipeline run "
+                "that this step run belongs to. Pipeline Run ID: "
+                f"`{self.pipeline_run_id}`."
+            )
 
         body = StepRunResponseBody(
             user=self.user.to_model() if self.user else None,
@@ -311,8 +298,8 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
         if include_metadata:
             metadata = StepRunResponseMetadata(
                 project=self.project.to_model(),
-                config=full_step_config.config,
-                spec=full_step_config.spec,
+                config=step.config,
+                spec=step.spec,
                 cache_key=self.cache_key,
                 code_hash=self.code_hash,
                 docstring=self.docstring,
@@ -335,6 +322,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
 
         return StepRunResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/schemas/trigger_schemas.py

@@ -239,6 +239,7 @@ class TriggerSchema(NamedSchema, table=True):
             )
         return TriggerResponse(
             id=self.id,
+            project_id=self.project_id,
             name=self.name,
             body=body,
             metadata=metadata,

zenml/zen_stores/template_utils.py

@@ -96,22 +96,26 @@ def generate_config_template(
     deployment_model = deployment.to_model(include_metadata=True)
 
     steps_configs = {
-        name: step.config.model_dump(
+        name: step.step_config_overrides.model_dump(
             include=set(StepConfigurationUpdate.model_fields),
             exclude={"name", "outputs"},
+            exclude_none=True,
+            exclude_defaults=True,
         )
         for name, step in deployment_model.step_configurations.items()
     }
 
     for config in steps_configs.values():
-        config["settings"].pop("docker", None)
+        config.get("settings", {}).pop("docker", None)
 
     pipeline_config = deployment_model.pipeline_configuration.model_dump(
         include=set(PipelineRunConfiguration.model_fields),
         exclude={"schedule", "build", "parameters"},
+        exclude_none=True,
+        exclude_defaults=True,
     )
 
-    pipeline_config["settings"].pop("docker", None)
+    pipeline_config.get("settings", {}).pop("docker", None)
 
     config_template = {
         "run_name": deployment_model.run_name_template,

{zenml_nightly-0.82.0.dev20250509.dist-info → zenml_nightly-0.82.0.dev20250511.dist-info}/METADATA

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

{zenml_nightly-0.82.0.dev20250509.dist-info → zenml_nightly-0.82.0.dev20250511.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=N0gO4bt86dg_dhEgiYxH8Ak0yGLW-basYq9JJugfjPA,19
+zenml/VERSION,sha256=sNYKOurs-d4b5GAaVuicbVQeHyNYJIL-DKO5dTE8P_E,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
@@ -65,11 +65,11 @@ zenml/code_repositories/local_repository_context.py,sha256=1VyiYkJBDVg0iGusgRQDT
 zenml/config/__init__.py,sha256=DZEic7euSbwI9Yb3FMRQhTgfhqz-C6OdAiYmOb0-opI,1519
 zenml/config/base_settings.py,sha256=itoLqc1cOwEYhgSGdZmSKSaBevQkvYH7NQh7PUamazc,1700
 zenml/config/build_configuration.py,sha256=tEAIXwAGS8Flxo4mK5BGHsBk7LPWW16MOa6oRDymZ7k,6172
-zenml/config/compiler.py,sha256=jzyyTmM8nTxtpIVTSjZ-jrVOBJ1roVoJ3fs1A1nwv9M,24130
+zenml/config/compiler.py,sha256=bK3LCDkrFc9SapJYH-vuQZ_o8scHNs-FdC5DblIUU4U,23024
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=w62SDW2Qo5jFxcikHr8p-Lk6ymC2ieh3MnEfkte9Lf0,13322
 zenml/config/global_config.py,sha256=ZD3WodfcWMBJZOl1FNn3ztzwilGDDv9EKdHClLqSO8s,29562
-zenml/config/pipeline_configurations.py,sha256=hHhAyol2kE2kiFISOO-TB7wBJOpUUt4Ubv-T1holzbI,3799
+zenml/config/pipeline_configurations.py,sha256=7trCbElpqGGgawii2FrdLW8fKaAWCR8jACkNqdG_vcQ,3983
 zenml/config/pipeline_run_configuration.py,sha256=Y9C5mVUH-w4gc1w1PCQFpjEmfBBpSMvb8riA_sL78hY,2311
 zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE,2821
 zenml/config/resource_settings.py,sha256=0taXGHvDfXuvpMplxsuzpznB1sAvWJIGnXoVJ9-ySfw,3899
@@ -80,12 +80,12 @@ zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wz
 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
+zenml/config/step_configurations.py,sha256=vjGtCoHneUJeaMkTiuzl7TTPBO94MefxNlFIAL1l5rU,13038
 zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I,1888
 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=VI_-MR69Qxx3kEgxyRWL4Tf3wNAQyPYuJwsp6Yk59_Y,16485
+zenml/constants.py,sha256=Lkm9EWkhAEwxdOuVHxSmLB6XEpa6-sw1TfKvpaVNEhk,16545
 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
@@ -624,13 +624,13 @@ zenml/models/v2/base/base.py,sha256=Oz6iWCaJw1ykoYe2tTvGJqalGmDp0tRS9r1NpBx8WsM,
 zenml/models/v2/base/base_plugin_flavor.py,sha256=BfPL4gm5i7ad7_vuRmPtC-rE2kl0W_8SmpWme8Akv1c,2608
 zenml/models/v2/base/filter.py,sha256=hNPUR_BJ0vomb9fgB8D64p8nHMUAvS7Izv3BHFQ6bh4,40834
 zenml/models/v2/base/page.py,sha256=yBpWB-8iUmyRwXKSJvX3v9SB7WOnyRa25O8SAss24xk,2601
-zenml/models/v2/base/scoped.py,sha256=28OEN7Y0zINvVSqD_ATpeVvAEjqw6aGCbvaO7N_3o4g,22499
+zenml/models/v2/base/scoped.py,sha256=XxBqIJnj8QYUjAdLE4uGIj1euG0BMbO9pmBTmbzdKgQ,22554
 zenml/models/v2/core/__init__.py,sha256=LGMIUJi19sOsvo54roZSQwDp_czNQYtenqd_frTLIhU,613
 zenml/models/v2/core/action.py,sha256=IsgL4y9MUqEEbSNRSx03J9Bnv1uurOwLq7V2Jz9Kjnc,7887
 zenml/models/v2/core/action_flavor.py,sha256=DlGlJTOcgLSDYAsrbAyXOJVdo7X2GYTFQTTEG5BEf-Q,1718
 zenml/models/v2/core/api_key.py,sha256=0iye9ftNk-Gjg4izZqYqbNgF1UR51WdRfasiAE2TqPw,11536
 zenml/models/v2/core/artifact.py,sha256=m8Tw_DNaum3P-jP7pV2GdHvvv0EbsLkpwEbUx9JDMNM,8084
-zenml/models/v2/core/artifact_version.py,sha256=PQRWWkOHztQVNUBDhe2pTcHr_kNMnYjYQXRh69zWqXk,23565
+zenml/models/v2/core/artifact_version.py,sha256=vkzVWN7cJ9go8D_vIHbKG-pn3TA7LSMbg-09tbxB67M,23631
 zenml/models/v2/core/artifact_visualization.py,sha256=milAHTpfuYd8Dx_oB4hcPYJL-TMoiU0FwUI0UhB3F-E,3115
 zenml/models/v2/core/code_reference.py,sha256=9xy3Gzychq16lLUs01va3gGfwdu-AMrFgys_ua89uFY,3557
 zenml/models/v2/core/code_repository.py,sha256=qMqHF4uU2LkSShdBKb14Sg3XLGHLd7aMUEOyc_H0uZs,5556
@@ -690,7 +690,7 @@ zenml/orchestrators/local_docker/__init__.py,sha256=k8J68ydy6HmmvE9tWo32g761H8P_
 zenml/orchestrators/local_docker/local_docker_orchestrator.py,sha256=4b4oiCQ_PnfzgsHZuGrWeVEvjsrq_0yvVqtBUqz3Ai4,9353
 zenml/orchestrators/output_utils.py,sha256=01vqke1ZfmfuLpgxNerF-QL2wA0VPv1zUdvlMw0OwUY,3508
 zenml/orchestrators/publish_utils.py,sha256=_77t2sVq3Tm-yAYg2NPxcCD7Y8lwe0lm55NNrtMUzQQ,4741
-zenml/orchestrators/step_launcher.py,sha256=b0TWQRjhEAMylKf6H4W29PSyOAoGQzc2J89RIQ3OkdY,17846
+zenml/orchestrators/step_launcher.py,sha256=YOeQtvmI__c8ugrqk2KSyITYVOxoLIAZxUIxkLyv-UY,17857
 zenml/orchestrators/step_run_utils.py,sha256=7GOWAW3izFopGPkvvjYyJKiYgYuHuHJ8V6eZfwjAx8E,13401
 zenml/orchestrators/step_runner.py,sha256=EUgKG_g0fOQ6gnB1hPSSa6UXwUKVkguC-Yj-Q0yEQXg,26632
 zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7cM,5784
@@ -701,7 +701,7 @@ zenml/pipelines/build_utils.py,sha256=FYflpfK8i9jLgy8Ybr2dhQZarUDA1_0shvEARPkBMY
 zenml/pipelines/pipeline_context.py,sha256=4BixReLcPo33VtNBDrMwnJqjKTinHjmO5AOfmoeIOQM,3659
 zenml/pipelines/pipeline_decorator.py,sha256=LB21QYrbFeBdUGwKBUNbdpXAxO4OOtYl5Vs_mzJNXqU,4600
 zenml/pipelines/pipeline_definition.py,sha256=954y2DmoY9luHhutC2nVRXzZ_KLnTWFnS5LSzrq_OZo,59637
-zenml/pipelines/run_utils.py,sha256=nY56k7ITTCkRt6ejPRKIXtSU8EIK1BEJb6w-4lR4QPA,12282
+zenml/pipelines/run_utils.py,sha256=uhccBmXcS_4GkIV33wom4R9bT_AKHr385DXnHAN8iGo,12315
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
 zenml/plugins/plugin_flavor_registry.py,sha256=LsN2Q0K-7EQ9H4uvlEG62Y0C1_Ro1UwppX4cnGbEcOA,10862
@@ -736,7 +736,7 @@ zenml/stack/authentication_mixin.py,sha256=_Rn6SurHnyBuU_CJBOkwPR305yQFfqN60t6nE
 zenml/stack/flavor.py,sha256=wFLjajCw_G2NMR2UJzEGH2dcLj1dvmU9dDI5iS8b9rk,10544
 zenml/stack/flavor_registry.py,sha256=IL0fRrxxQJ9YkCYCeADP7nwWEQo4XBElJY4owMjKGbQ,6108
 zenml/stack/stack.py,sha256=3mLGNBuep0PeBcoFfOEaB6Ya206yav1ppiH5YEjp9xA,33042
-zenml/stack/stack_component.py,sha256=hPlAOpNTUteoCkicMHEnHFOXPMbMLzPhxKN0Kt_CLWk,29244
+zenml/stack/stack_component.py,sha256=D6rSPvbOIiTCX0k1iLfdnmPwycPf2jnnkcMY2xCnZiA,29327
 zenml/stack/stack_validator.py,sha256=hWbvvGIeWLj6NwSsF4GCc6RAxAWvxHXTcBZL9nJvcak,3111
 zenml/stack/utils.py,sha256=CHs9rxdqHkUT12KPhJX1YPtIWnZBoVlRlq5PzNymq3E,6406
 zenml/stack_deployments/__init__.py,sha256=-7593cQ_ZgRn774Ol-8AKXXQquIU4DSiaThVEr6TfWM,644
@@ -1024,7 +1024,7 @@ zenml/zen_server/rbac/endpoint_utils.py,sha256=l6V6DwsPGmT6dLJmY4MHMRfRXdc0lZVib
 zenml/zen_server/rbac/models.py,sha256=Xhz0XqmVqg3ji9uTwjlhQ4IuQ2ivrT4gVdHi5ZkjFC4,4933
 zenml/zen_server/rbac/rbac_interface.py,sha256=VPMNnUIPcqJWDngITX6cfaPlM6zJwzCApXAnT_oaxSQ,3431
 zenml/zen_server/rbac/rbac_sql_zen_store.py,sha256=djGycBkciChk48q8oL7mHM0c-cHJM-S7inv5f3gcDgg,5942
-zenml/zen_server/rbac/utils.py,sha256=8hBvCGWnz517hL5gn53dcjcrKePE2OGqZV_ojJ8XZEw,24329
+zenml/zen_server/rbac/utils.py,sha256=09r1dSZ04hcag21rg6Ug2fMCdYfNjSBnx_W8C336GzE,24254
 zenml/zen_server/rbac/zenml_cloud_rbac.py,sha256=L1Tio4IcNNxNWnVpAqBWYgdVfrqR20ddvdhWAo1i3f4,6055
 zenml/zen_server/routers/__init__.py,sha256=ViyAhWL-ogHxE9wBvB_iMcur5H1NRVrzXkpogVY7FBA,641
 zenml/zen_server/routers/actions_endpoints.py,sha256=MFausi27i5QBB5s4vbjQQTVqFiRlk5jF5lRM6vsRM2o,9422
@@ -1039,7 +1039,7 @@ zenml/zen_server/routers/logs_endpoints.py,sha256=qq7vLfJzmjIZ9C4DSw_IPjqHi5LdPb
 zenml/zen_server/routers/model_versions_endpoints.py,sha256=VhmBUuMFef8bXvDIRncPD5hLa851lmIvmASI_lw2F5o,15483
 zenml/zen_server/routers/models_endpoints.py,sha256=Bh_f7wvw1ZNwmZ51LRiiz3ftatkp_J5t622bJinmOaY,5680
 zenml/zen_server/routers/pipeline_builds_endpoints.py,sha256=PQRSrDcyor6laZyOl74BODcdg0JSdFr-hbNvEJkJIG0,5333
-zenml/zen_server/routers/pipeline_deployments_endpoints.py,sha256=-art7WP8X1LzSaPPDZe484fZWLcJhnutltTiK5bx73s,7361
+zenml/zen_server/routers/pipeline_deployments_endpoints.py,sha256=f45O7dQqZnFWA7C62N5VGAtBsqpt38cRYsGgTaBS3Dw,9534
 zenml/zen_server/routers/pipelines_endpoints.py,sha256=R6OM-0Tc6ZNPzQmLs5MWTJIE4FaksHt6xuDqCBtmr1U,7782
 zenml/zen_server/routers/plugin_endpoints.py,sha256=vgIdA0kdYsf8PeF_pqDaqEYnIi2mlYALAv4uB9Nn7S0,3239
 zenml/zen_server/routers/projects_endpoints.py,sha256=iIgCg_0_iR5OddgULvsP4xxMOOmEGmtCuSKYZVRE9f4,8464
@@ -1064,7 +1064,7 @@ 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=jx_FdkrXKI-LOG1DykedhpNPwoUm_iFDJF9r1UVaE_o,19680
+zenml/zen_server/template_execution/utils.py,sha256=6a62U0hLDe79xLbZM-WbzY-miOeTejQ7vpIkmch0ggU,19086
 zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8ajuZE01DaHmdCDCZmsroDcTarvN-nE8jv6qQ,2590
 zenml/zen_server/utils.py,sha256=yOwPI0DxzDIxF7Ppnty_DuGMWYJ_ubHUNWfbTQ1_q9Q,18461
 zenml/zen_server/zen_server_api.py,sha256=1_YHOUmBulXuHWCxDLYtDCMxGze159UmU3f05643PSg,18182
@@ -1274,36 +1274,36 @@ zenml/zen_stores/migrations/versions/fbd7f18ced1e_increase_step_run_field_length
 zenml/zen_stores/migrations/versions/ff538a321a92_migrate_onboarding_state.py,sha256=gsUFLJQ32_o9U35JCVqkqJVVk-zfq3yel25hXhzVFm4,3829
 zenml/zen_stores/rest_zen_store.py,sha256=-Xjw9QTY6lvgp5mDE4bBIhU_ANADS83Si88xMcLTFIE,158371
 zenml/zen_stores/schemas/__init__.py,sha256=4EXqExiVyxdnGxhQ_Hz79mOdRuMD0LsGlw0PaP2Ef6o,4333
-zenml/zen_stores/schemas/action_schemas.py,sha256=2OiUiskFSg5qXGxA6AFq71bWzUczxA563LGFokLZmac,6456
+zenml/zen_stores/schemas/action_schemas.py,sha256=sv2J2TP12MeyGPQR2JsOPIivbPQ5OImg64exYS7CZBM,6496
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=0pK7b9HlJuQL3DuKT4eGjFb87tyd4x-E2VyxJLpRv3o,7459
-zenml/zen_stores/schemas/artifact_schemas.py,sha256=alVpqfaovnZLy4Xl0kPICVOd9OIQGUSftpirRTzMktg,16008
+zenml/zen_stores/schemas/artifact_schemas.py,sha256=wefj8m1RUeTrfU3rJwFYKYkdJq5Ycpqxm_zbbFrOCvI,16088
 zenml/zen_stores/schemas/artifact_visualization_schemas.py,sha256=zv_Mg--WMTtAPEy5PXIkZlxq-oLrXZNLHXb_YgZbm0g,3667
 zenml/zen_stores/schemas/base_schemas.py,sha256=-MDYnS4ZLDe6x75nTXuVrOSbcYOuECLNk-yxjfEdJh4,2043
-zenml/zen_stores/schemas/code_repository_schemas.py,sha256=cJh4LMtGGSEsZftUYVJIrwSDpYZrTyxQw5k3AJYRTX0,7622
+zenml/zen_stores/schemas/code_repository_schemas.py,sha256=24g-ZyuW9buDhGTJ_5B2WXVv6waBAtwH6X1DNdIzEbA,7662
 zenml/zen_stores/schemas/component_schemas.py,sha256=sk_xs0OGInStSJltMPbVaktnVUk_oC5oX0R3ClkPeS8,7948
 zenml/zen_stores/schemas/constants.py,sha256=bcadtiWEWkpzRzvnj46yuAaC7DE09g2H9L9r398lV00,704
 zenml/zen_stores/schemas/device_schemas.py,sha256=voVBknm14NmFpij1EGb_HhVlUng0cglCMXkYSxDRq1s,9126
-zenml/zen_stores/schemas/event_source_schemas.py,sha256=st706lmxNiQx0qT7jBFApbqogJC7KuwnsRuQ2DCpxy8,6555
+zenml/zen_stores/schemas/event_source_schemas.py,sha256=Psj2xcPLN30MEDiihrEahJGh4jL4N7XwxzXvPDzbQRw,6595
 zenml/zen_stores/schemas/flavor_schemas.py,sha256=jlEhhhpqvhzsdwShE_J3mJ7sqgfRKou6KKAh6bOvhkM,4690
 zenml/zen_stores/schemas/logs_schemas.py,sha256=qv6fs3JiVgzlmTXJqb_gG5NsU5q_50e0167JiWIxR14,3588
-zenml/zen_stores/schemas/model_schemas.py,sha256=5QqdQYXoe-XjFBg-LtugJG8314avcqZRD7RKDhmrHgs,25500
-zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=O13x8VPPZ3FirX9wkRLZD3IdvQpfohh4935qxXjgf24,5842
-zenml/zen_stores/schemas/pipeline_deployment_schemas.py,sha256=rJEnDbHqFwLfdJmEm2wZ6Gy2uxpaQ5Gzo8dt8i3GSTM,10867
-zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=XDwLVYhnXGEhLSwAXLooJEf_EHzPQ64xvC2Hzj4aoWs,18681
-zenml/zen_stores/schemas/pipeline_schemas.py,sha256=oC0A4zE22DXiX3jl7fDM7dHTTbwQcfvemC8OaX5TGg4,7133
+zenml/zen_stores/schemas/model_schemas.py,sha256=RpW78rHclJ7hyq789F6R6VUrluPWLvtpfpBTTgwGx5Y,25580
+zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=zAF8OVAAPgAeg-pmihlayIljwiAj3eoF7GVCCGNnP68,5882
+zenml/zen_stores/schemas/pipeline_deployment_schemas.py,sha256=FBXkV19_KhpjRYTkx3QkfK07WdVTbjkiotydsQVS71I,11253
+zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=quxso7FC-VxxBiJyrDjpDsE-ZtfDxWw2XIay-cla1Vg,18608
+zenml/zen_stores/schemas/pipeline_schemas.py,sha256=5m4IyFu8bSF4kqMIeu9cJl04mjlAoAqX6CMLUQoFCp4,7173
 zenml/zen_stores/schemas/project_schemas.py,sha256=X2GClPNQz0COsEZX8xI-I8Sm68Hb6f20Obm24mQyLS0,6013
 zenml/zen_stores/schemas/run_metadata_schemas.py,sha256=G94rT4ldluMSnf9rm7R_9rw_GlgaAyq72ptkHl0gHeg,3605
-zenml/zen_stores/schemas/run_template_schemas.py,sha256=sM3pcegx7qOs5tlwqVnFvc7JmWcSPOFxmk7LAndnwdY,10043
-zenml/zen_stores/schemas/schedule_schema.py,sha256=E9hmq8-PBRHxEH5UYEpD_Oc2cHkNp4FG0sTvhiLZ6D4,7653
+zenml/zen_stores/schemas/run_template_schemas.py,sha256=WVi_B4wFzoCtHOsRhb8qOONUih1vGGxanviefW8uKKg,10083
+zenml/zen_stores/schemas/schedule_schema.py,sha256=DdVXlNn62tg5m2gqKAuWhIvvl8y-MQBjIplUapYPvQQ,7693
 zenml/zen_stores/schemas/schema_utils.py,sha256=Xahifq2fJ5szXCM00ZZ6461Div9Suatzl6sy9hVhPkk,3612
 zenml/zen_stores/schemas/secret_schemas.py,sha256=HrihLSUPJKmWbQHogGmi6e0lYPoQHUGWGRl69sLnZMk,9430
 zenml/zen_stores/schemas/server_settings_schemas.py,sha256=usBE-idRrmK-LeLN0zDtCRCGP51YTnyKfIx5GZ0_ATg,4275
 zenml/zen_stores/schemas/service_connector_schemas.py,sha256=kNJ6TmM2RA16wc73zusosHxWUm6gVpyowlY_OJv-Oww,9709
-zenml/zen_stores/schemas/service_schemas.py,sha256=yg9OJ2QWa1qQy_gqraGvN3JQrOJDLjhkSL-UVSFJy-s,9517
+zenml/zen_stores/schemas/service_schemas.py,sha256=X9Z8yYM6AmOC69EO8kmWDU9EnO_KW10CXrQzM6EQ2Rc,9557
 zenml/zen_stores/schemas/stack_schemas.py,sha256=SkMspxsnR420KU4mZlulPbEOR4DMSRyrcspe3HSYjHc,5393
-zenml/zen_stores/schemas/step_run_schemas.py,sha256=NuJtCXZ_Zacz4_X7GZ5PQ-fLpzVBhnhScGx2fBqAlDg,16333
+zenml/zen_stores/schemas/step_run_schemas.py,sha256=xZc7ABtwcCdzulw06_OHLva2XGRl_NI7HjeuN1ae71M,15762
 zenml/zen_stores/schemas/tag_schemas.py,sha256=GMKdyT5pZb977vscOW41N2beVt-IHcELSiSSIBHqlUI,6377
-zenml/zen_stores/schemas/trigger_schemas.py,sha256=IOfd6enpAr9mwN5p4KbA2jvQEHlMEUXDXAMqIULS4lc,10711
+zenml/zen_stores/schemas/trigger_schemas.py,sha256=KfEQlXZh537cieEwvqY9g4VifG7PfAdll_BPZ5Q9xBA,10751
 zenml/zen_stores/schemas/user_schemas.py,sha256=wLSHtfjjrYBWnX_PXv5KPz_DRdheTxZVjcgBe-cP__c,11497
 zenml/zen_stores/schemas/utils.py,sha256=82-LQj1EV4jp69kKE2GolQw_zTDBvAkyA4cHDaSIj6U,4160
 zenml/zen_stores/secrets_stores/__init__.py,sha256=VGMVerFv3xUWxnp5X_bHyQJwGCeN7BTovUDwho_w-4w,651
@@ -1316,10 +1316,10 @@ zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
 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/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.82.0.dev20250509.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.82.0.dev20250509.dist-info/METADATA,sha256=hp3yQJ8a4QIIXoLyCUlSkrCGaQs3SG09N2E4s_EEsCY,24315
-zenml_nightly-0.82.0.dev20250509.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.82.0.dev20250509.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.82.0.dev20250509.dist-info/RECORD,,
+zenml_nightly-0.82.0.dev20250511.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.82.0.dev20250511.dist-info/METADATA,sha256=d-G4MWVNQJVUJbr34gl5sEHO-l5rUjoR09KrF0PaRHU,24315
+zenml_nightly-0.82.0.dev20250511.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.82.0.dev20250511.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.82.0.dev20250511.dist-info/RECORD,,