zenml-nightly 0.84.3.dev20250903__py3-none-any.whl → 0.84.3.dev20250906__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.84.3.dev20250903
+0.84.3.dev20250906

zenml/artifacts/utils.py

@@ -178,6 +178,8 @@ def _store_artifact_data_and_prepare_request(
         # the materializer
         combined_metadata.update(metadata or {})
 
+    content_hash = materializer.compute_content_hash(data)
+
     artifact_version_request = ArtifactVersionRequest(
         artifact_name=name,
         version=version,
@@ -186,6 +188,7 @@ def _store_artifact_data_and_prepare_request(
         uri=materializer.uri,
         materializer=source_utils.resolve(materializer.__class__),
         data_type=source_utils.resolve(data_type),
+        content_hash=content_hash,
         project=Client().active_project.id,
         artifact_store_id=artifact_store.id,
         visualizations=visualizations,

zenml/config/__init__.py

@@ -11,18 +11,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-"""The `config` module contains classes and functions that manage user-specific configuration.
-
-ZenML's configuration is stored in a file called
-``config.yaml``, located on the user's directory for configuration files.
-(The exact location differs from operating system to operating system.)
-
-The ``GlobalConfiguration`` class is the main class in this module. It provides
-a Pydantic configuration object that is used to store and retrieve
-configuration. This ``GlobalConfiguration`` object handles the serialization and
-deserialization of the configuration options that are stored in the file in
-order to persist the configuration across sessions.
-"""
+"""Config classes."""
 from zenml.config.docker_settings import (
     DockerSettings,
     PythonPackageInstaller,
@@ -32,6 +21,7 @@ from zenml.config.resource_settings import ResourceSettings, ByteUnit
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.schedule import Schedule
 from zenml.config.store_config import StoreConfiguration
+from zenml.config.cache_policy import CachePolicy
 
 __all__ = [
     "DockerSettings",
@@ -42,4 +32,5 @@ __all__ = [
     "StepRetryConfig",
     "Schedule",
     "StoreConfiguration",
+    "CachePolicy",
 ]

zenml/config/cache_policy.py

@@ -0,0 +1,104 @@
+#  Copyright (c) ZenML GmbH 2025. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at:
+#
+#       https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+#  or implied. See the License for the specific language governing
+#  permissions and limitations under the License.
+"""Cache policy."""
+
+from typing import Any, List, Optional, Union
+
+from pydantic import BaseModel, BeforeValidator, Field
+from typing_extensions import Annotated
+
+from zenml.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+class CachePolicy(BaseModel):
+    """Cache policy."""
+
+    include_step_code: bool = Field(
+        default=True,
+        description="Whether to include the step code in the cache key.",
+    )
+    include_step_parameters: bool = Field(
+        default=True,
+        description="Whether to include the step parameters in the cache key.",
+    )
+    include_artifact_values: bool = Field(
+        default=True,
+        description="Whether to include the artifact values in the cache key. "
+        "If the materializer for an artifact doesn't support generating a "
+        "content hash, the artifact ID will be used as a fallback if enabled.",
+    )
+    include_artifact_ids: bool = Field(
+        default=True,
+        description="Whether to include the artifact IDs in the cache key.",
+    )
+    ignored_inputs: Optional[List[str]] = Field(
+        default=None,
+        description="List of input names to ignore in the cache key.",
+    )
+
+    @classmethod
+    def default(cls) -> "CachePolicy":
+        """Default cache policy.
+
+        Returns:
+            The default cache policy.
+        """
+        return cls(
+            include_step_code=True,
+            include_step_parameters=True,
+            include_artifact_values=True,
+            include_artifact_ids=True,
+            ignored_inputs=None,
+        )
+
+    @classmethod
+    def from_string(cls, value: str) -> "CachePolicy":
+        """Create a cache policy from a string.
+
+        Args:
+            value: The string to create a cache policy from.
+
+        Raises:
+            ValueError: If the string is not a valid cache policy.
+
+        Returns:
+            The cache policy.
+        """
+        if value.lower() == "default":
+            return cls.default()
+        else:
+            raise ValueError(f"Invalid cache policy: {value}")
+
+
+def _convert_cache_policy(value: Any) -> Any:
+    """Converts a potential cache policy string to a cache policy object.
+
+    Args:
+        value: The value to convert.
+
+    Returns:
+        The converted value.
+    """
+    if isinstance(value, str):
+        return CachePolicy.from_string(value)
+
+    return value
+
+
+CachePolicyWithValidator = Annotated[
+    CachePolicy, BeforeValidator(_convert_cache_policy)
+]
+CachePolicyOrString = Union[CachePolicy, str]

zenml/config/compiler.py

@@ -208,6 +208,7 @@ class Compiler:
                 model=config.model,
                 retry=config.retry,
                 parameters=config.parameters,
+                cache_policy=config.cache_policy,
             )
 
         invalid_step_configs = set(config.steps) - set(pipeline.invocations)

zenml/config/pipeline_configurations.py

@@ -18,6 +18,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from pydantic import SerializeAsAny, field_validator
 
+from zenml.config.cache_policy import CachePolicyWithValidator
 from zenml.config.constants import DOCKER_SETTINGS_KEY
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.source import SourceWithValidator
@@ -51,6 +52,7 @@ class PipelineConfigurationUpdate(StrictBaseModel):
     parameters: Optional[Dict[str, Any]] = None
     retry: Optional[StepRetryConfig] = None
     substitutions: Dict[str, str] = {}
+    cache_policy: Optional[CachePolicyWithValidator] = None
 
     def finalize_substitutions(
         self, start_time: Optional[datetime] = None, inplace: bool = False

zenml/config/pipeline_run_configuration.py

@@ -19,6 +19,7 @@ from uuid import UUID
 from pydantic import Field, SerializeAsAny
 
 from zenml.config.base_settings import BaseSettings
+from zenml.config.cache_policy import CachePolicyWithValidator
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.schedule import Schedule
 from zenml.config.source import SourceWithValidator
@@ -55,3 +56,4 @@ class PipelineRunConfiguration(
     failure_hook_source: Optional[SourceWithValidator] = None
     success_hook_source: Optional[SourceWithValidator] = None
     substitutions: Dict[str, str] = {}
+    cache_policy: Optional[CachePolicyWithValidator] = None

zenml/config/step_configurations.py

@@ -37,6 +37,7 @@ from zenml.artifacts.external_artifact_config import (
 )
 from zenml.client_lazy_loader import ClientLazyLoader
 from zenml.config.base_settings import BaseSettings, SettingsOrDict
+from zenml.config.cache_policy import CachePolicy, CachePolicyWithValidator
 from zenml.config.constants import DOCKER_SETTINGS_KEY, RESOURCE_SETTINGS_KEY
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.source import Source, SourceWithValidator
@@ -155,6 +156,7 @@ class StepConfigurationUpdate(StrictBaseModel):
     model: Optional[Model] = None
     retry: Optional[StepRetryConfig] = None
     substitutions: Dict[str, str] = {}
+    cache_policy: Optional[CachePolicyWithValidator] = None
 
     outputs: Mapping[str, PartialArtifactConfiguration] = {}
 
@@ -200,6 +202,7 @@ class PartialStepConfiguration(StepConfigurationUpdate):
     model_artifacts_or_metadata: Mapping[str, ModelVersionDataLazyLoader] = {}
     client_lazy_loaders: Mapping[str, ClientLazyLoader] = {}
     outputs: Mapping[str, PartialArtifactConfiguration] = {}
+    cache_policy: CachePolicyWithValidator = CachePolicy.default()
 
     # TODO: In Pydantic v2, the `model_` is a protected namespaces for all
     #  fields defined under base models. If not handled, this raises a warning.
@@ -291,6 +294,7 @@ class StepConfiguration(PartialStepConfiguration):
                 "success_hook_source",
                 "retry",
                 "substitutions",
+                "cache_policy",
             },
             exclude_none=True,
         )
@@ -303,6 +307,7 @@ class StepConfiguration(PartialStepConfiguration):
                     "success_hook_source",
                     "retry",
                     "substitutions",
+                    "cache_policy",
                 },
                 exclude_none=True,
             )

zenml/integrations/airflow/orchestrators/dag_generator.py

@@ -225,7 +225,7 @@ else:
         dag_id=dag_config.id,
         is_paused_upon_creation=False,
         tags=dag_config.tags,
-        schedule_interval=dag_config.schedule,
+        schedule=dag_config.schedule,
         start_date=dag_config.start_date,
         end_date=dag_config.end_date,
         catchup=dag_config.catchup,

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py

@@ -57,7 +57,11 @@ from zenml.integrations.kubernetes.orchestrators.manifest_utils import (
 )
 from zenml.logger import get_logger
 from zenml.logging.step_logging import setup_orchestrator_logging
-from zenml.models import PipelineDeploymentResponse, PipelineRunResponse
+from zenml.models import (
+    PipelineDeploymentResponse,
+    PipelineRunResponse,
+    PipelineRunUpdate,
+)
 from zenml.orchestrators import publish_utils
 from zenml.orchestrators.step_run_utils import (
     StepRunRequestFactory,
@@ -257,7 +261,12 @@ def main() -> None:
     else:
         orchestrator_run_id = orchestrator_pod_name
         if args.run_id:
-            pipeline_run = client.get_pipeline_run(args.run_id)
+            pipeline_run = client.zen_store.update_run(
+                run_id=args.run_id,
+                run_update=PipelineRunUpdate(
+                    orchestrator_run_id=orchestrator_run_id
+                ),
+            )
         else:
             pipeline_run = create_placeholder_run(
                 deployment=deployment,

zenml/materializers/base_materializer.py

@@ -229,6 +229,17 @@ class BaseMaterializer(metaclass=BaseMaterializerMeta):
         # Optionally, extract some metadata from `data` for ZenML to store.
         return {}
 
+    def compute_content_hash(self, data: Any) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        return None
+
     # ================
     # Internal Methods
     # ================

zenml/materializers/built_in_materializer.py

@@ -13,6 +13,8 @@
 #  permissions and limitations under the License.
 """Implementation of ZenML's builtin materializer."""
 
+import hashlib
+import json
 import os
 from typing import (
     TYPE_CHECKING,
@@ -138,6 +140,20 @@ class BuiltInMaterializer(BaseMaterializer):
 
         return {}
 
+    def compute_content_hash(self, data: Any) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        hash_ = hashlib.md5(usedforsecurity=False)
+        hash_.update(self.__class__.__name__.encode())
+        hash_.update(json.dumps(data, sort_keys=True).encode())
+        return hash_.hexdigest()
+
 
 class BytesMaterializer(BaseMaterializer):
     """Handle `bytes` data type, which is not JSON serializable."""
@@ -189,6 +205,20 @@ class BytesMaterializer(BaseMaterializer):
         """
         return {self.data_path.replace("\\", "/"): VisualizationType.MARKDOWN}
 
+    def compute_content_hash(self, data: Any) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        hash_ = hashlib.md5(usedforsecurity=False)
+        hash_.update(self.__class__.__name__.encode())
+        hash_.update(data)
+        return hash_.hexdigest()
+
 
 def _all_serializable(iterable: Iterable[Any]) -> bool:
     """For an iterable, check whether all of its elements are JSON-serializable.
@@ -224,6 +254,21 @@ def _is_serializable(obj: Any) -> bool:
     return False
 
 
+def _custom_json_converter(obj: Any) -> Any:
+    """Custom JSON converter that handles sets and tuples.
+
+    Args:
+        obj: The object to convert.
+
+    Returns:
+        The converted object.
+    """
+    if isinstance(obj, (set, tuple)):
+        return list(obj)
+
+    return obj
+
+
 def find_type_by_str(type_str: str) -> Type[Any]:
     """Get a Python type, given its string representation.
 
@@ -489,3 +534,24 @@ class BuiltInContainerMaterializer(BaseMaterializer):
         if hasattr(data, "__len__"):
             return {"length": len(data)}
         return {}
+
+    def compute_content_hash(self, data: Any) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        if _is_serializable(data):
+            hash_ = hashlib.md5(usedforsecurity=False)
+            hash_.update(self.__class__.__name__.encode())
+            hash_.update(
+                json.dumps(
+                    data, sort_keys=True, default=_custom_json_converter
+                ).encode()
+            )
+            return hash_.hexdigest()
+
+        return None

zenml/materializers/pydantic_materializer.py

@@ -13,8 +13,10 @@
 #  permissions and limitations under the License.
 """Implementation of ZenML's pydantic materializer."""
 
+import hashlib
+import json
 import os
-from typing import TYPE_CHECKING, Any, ClassVar, Dict, Tuple, Type
+from typing import TYPE_CHECKING, Any, ClassVar, Dict, Optional, Tuple, Type
 
 from pydantic import BaseModel
 
@@ -66,3 +68,21 @@ class PydanticMaterializer(BaseMaterializer):
             The extracted metadata as a dictionary.
         """
         return {"schema": data.schema()}
+
+    def compute_content_hash(self, data: BaseModel) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        hash_ = hashlib.md5(usedforsecurity=False)
+        hash_.update(self.__class__.__name__.encode())
+
+        json_data = data.model_dump(mode="json")
+        hash_.update(json.dumps(json_data, sort_keys=True).encode())
+        return hash_.hexdigest()
+
+        return None

zenml/materializers/structured_string_materializer.py

@@ -13,8 +13,9 @@
 #  permissions and limitations under the License.
 """Implementation of HTMLString materializer."""
 
+import hashlib
 import os
-from typing import Dict, Type, Union
+from typing import Dict, Optional, Type, Union
 
 from zenml.enums import ArtifactType, VisualizationType
 from zenml.logger import get_logger
@@ -129,3 +130,17 @@ class StructuredStringMaterializer(BaseMaterializer):
             raise ValueError(
                 f"Data type {data_type} is not supported by this materializer."
             )
+
+    def compute_content_hash(self, data: STRUCTURED_STRINGS) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        hash_ = hashlib.md5(usedforsecurity=False)
+        hash_.update(self.__class__.__name__.encode())
+        hash_.update(data.encode())
+        return hash_.hexdigest()

zenml/materializers/uuid_materializer.py

@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 """Implementation of ZenML's UUID materializer."""
 
+import hashlib
 import os
 import uuid
 from typing import Any, ClassVar, Dict, Optional, Tuple, Type
@@ -77,3 +78,17 @@ class UUIDMaterializer(BaseMaterializer):
         return {
             "string_representation": str(data),
         }
+
+    def compute_content_hash(self, data: uuid.UUID) -> Optional[str]:
+        """Compute the content hash of the given data.
+
+        Args:
+            data: The data to compute the content hash of.
+
+        Returns:
+            The content hash of the given data.
+        """
+        hash_ = hashlib.md5(usedforsecurity=False)
+        hash_.update(self.__class__.__name__.encode())
+        hash_.update(data.bytes)
+        return hash_.hexdigest()

zenml/models/v2/core/artifact_version.py

@@ -104,6 +104,11 @@ class ArtifactVersionRequest(ProjectScopedRequest):
     data_type: SourceWithValidator = Field(
         title="Data type of the artifact.",
     )
+    content_hash: Optional[str] = Field(
+        title="The content hash of the artifact version.",
+        default=None,
+        max_length=STR_FIELD_MAX_LENGTH,
+    )
     tags: Optional[List[str]] = Field(
         title="Tags of the artifact.",
         description="Should be a list of plain strings, e.g., ['tag1', 'tag2']",
@@ -201,6 +206,10 @@ class ArtifactVersionResponseBody(ProjectScopedResponseBody):
         title="ID of the artifact store in which this artifact is stored.",
         default=None,
     )
+    content_hash: Optional[str] = Field(
+        title="The content hash of the artifact version.",
+        default=None,
+    )
 
     @field_validator("version")
     @classmethod
@@ -307,6 +316,15 @@ class ArtifactVersionResponse(
         """
         return self.get_body().type
 
+    @property
+    def content_hash(self) -> Optional[str]:
+        """The `content_hash` property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_body().content_hash
+
     @property
     def tags(self) -> List[TagResponse]:
         """The `tags` property.

zenml/models/v2/core/pipeline_run.py

@@ -144,6 +144,7 @@ class PipelineRunUpdate(BaseUpdate):
 
     status: Optional[ExecutionStatus] = None
     end_time: Optional[datetime] = None
+    orchestrator_run_id: Optional[str] = None
     # TODO: we should maybe have a different update model here, the upper
     #  three attributes should only be for internal use
     add_tags: Optional[List[str]] = Field(

zenml/orchestrators/base_orchestrator.py

@@ -35,7 +35,11 @@ from zenml.constants import (
     handle_bool_env_var,
 )
 from zenml.enums import ExecutionStatus, StackComponentType
-from zenml.exceptions import RunMonitoringError, RunStoppedException
+from zenml.exceptions import (
+    IllegalOperationError,
+    RunMonitoringError,
+    RunStoppedException,
+)
 from zenml.logger import get_logger
 from zenml.metadata.metadata_types import MetadataType
 from zenml.orchestrators.publish_utils import (
@@ -485,6 +489,7 @@ class BaseOrchestrator(StackComponent, ABC):
         Raises:
             NotImplementedError: If any orchestrator inheriting from the base
                 class does not implement this logic.
+            IllegalOperationError: If the run has no orchestrator run id yet.
         """
         # Check if the orchestrator supports cancellation
         if (
@@ -496,6 +501,12 @@ class BaseOrchestrator(StackComponent, ABC):
                 "support stopping pipeline runs."
             )
 
+        if not run.orchestrator_run_id:
+            raise IllegalOperationError(
+                "Cannot stop a pipeline run that has no orchestrator run id "
+                "yet."
+            )
+
         # Update pipeline status to STOPPING before calling concrete implementation
         publish_pipeline_run_status_update(
             pipeline_run_id=run.id,

zenml/orchestrators/cache_utils.py

@@ -14,9 +14,10 @@
 """Utilities for caching."""
 
 import hashlib
-from typing import TYPE_CHECKING, Dict, Optional
+from typing import TYPE_CHECKING, Mapping, Optional
 
 from zenml.client import Client
+from zenml.constants import CODE_HASH_PARAMETER_NAME
 from zenml.enums import ExecutionStatus, SorterOps
 from zenml.logger import get_logger
 from zenml.orchestrators import step_run_utils
@@ -27,6 +28,7 @@ if TYPE_CHECKING:
     from zenml.artifact_stores import BaseArtifactStore
     from zenml.config.step_configurations import Step
     from zenml.models import (
+        ArtifactVersionResponse,
         PipelineDeploymentResponse,
         PipelineRunResponse,
         StepRunResponse,
@@ -39,7 +41,7 @@ logger = get_logger(__name__)
 
 def generate_cache_key(
     step: "Step",
-    input_artifact_ids: Dict[str, "UUID"],
+    input_artifacts: Mapping[str, "ArtifactVersionResponse"],
     artifact_store: "BaseArtifactStore",
     project_id: "UUID",
 ) -> str:
@@ -60,13 +62,14 @@ def generate_cache_key(
 
     Args:
         step: The step to generate the cache key for.
-        input_artifact_ids: The input artifact IDs for the step.
+        input_artifacts: The input artifacts for the step.
         artifact_store: The artifact store of the active stack.
         project_id: The ID of the active project.
 
     Returns:
         A cache key.
     """
+    cache_policy = step.config.cache_policy
     hash_ = hashlib.md5()  # nosec
 
     # Project ID
@@ -85,15 +88,25 @@ def generate_cache_key(
     # when committing some unrelated files
     hash_.update(step.spec.source.import_path.encode())
 
-    # Step parameters
-    for key, value in sorted(step.config.parameters.items()):
-        hash_.update(key.encode())
-        hash_.update(str(value).encode())
+    if cache_policy.include_step_parameters:
+        for key, value in sorted(step.config.parameters.items()):
+            hash_.update(key.encode())
+            hash_.update(str(value).encode())
 
     # Input artifacts
-    for name, artifact_version_id in input_artifact_ids.items():
+    for name, artifact_version in input_artifacts.items():
+        if name in (cache_policy.ignored_inputs or []):
+            continue
+
         hash_.update(name.encode())
-        hash_.update(artifact_version_id.bytes)
+
+        if (
+            artifact_version.content_hash
+            and cache_policy.include_artifact_values
+        ):
+            hash_.update(artifact_version.content_hash.encode())
+        elif cache_policy.include_artifact_ids:
+            hash_.update(artifact_version.id.bytes)
 
     # Output artifacts and materializers
     for name, output in step.config.outputs.items():
@@ -103,6 +116,12 @@ def generate_cache_key(
 
     # Custom caching parameters
     for key, value in sorted(step.config.caching_parameters.items()):
+        if (
+            key == CODE_HASH_PARAMETER_NAME
+            and not cache_policy.include_step_code
+        ):
+            continue
+
         hash_.update(key.encode())
         hash_.update(str(value).encode())
 

zenml/orchestrators/step_run_utils.py

@@ -117,10 +117,6 @@ class StepRunRequestFactory:
             pipeline_run=self.pipeline_run,
             step_runs=step_runs,
         )
-        input_artifact_ids = {
-            input_name: artifact.id
-            for input_name, artifact in input_artifacts.items()
-        }
 
         request.inputs = {
             name: [artifact.id] for name, artifact in input_artifacts.items()
@@ -128,7 +124,7 @@ class StepRunRequestFactory:
 
         cache_key = cache_utils.generate_cache_key(
             step=step,
-            input_artifact_ids=input_artifact_ids,
+            input_artifacts=input_artifacts,
             artifact_store=self.stack.artifact_store,
             project_id=Client().active_project.id,
         )

zenml/pipelines/pipeline_decorator.py

@@ -29,6 +29,7 @@ from zenml.logger import get_logger
 
 if TYPE_CHECKING:
     from zenml.config.base_settings import SettingsOrDict
+    from zenml.config.cache_policy import CachePolicyOrString
     from zenml.config.retry_config import StepRetryConfig
     from zenml.model.model import Model
     from zenml.pipelines.pipeline_definition import Pipeline
@@ -60,6 +61,7 @@ def pipeline(
     model: Optional["Model"] = None,
     retry: Optional["StepRetryConfig"] = None,
     substitutions: Optional[Dict[str, str]] = None,
+    cache_policy: Optional["CachePolicyOrString"] = None,
 ) -> Callable[["F"], "Pipeline"]: ...
 
 
@@ -79,6 +81,7 @@ def pipeline(
     model: Optional["Model"] = None,
     retry: Optional["StepRetryConfig"] = None,
     substitutions: Optional[Dict[str, str]] = None,
+    cache_policy: Optional["CachePolicyOrString"] = None,
 ) -> Union["Pipeline", Callable[["F"], "Pipeline"]]:
     """Decorator to create a pipeline.
 
@@ -102,6 +105,7 @@ def pipeline(
         model: configuration of the model in the Model Control Plane.
         retry: Retry configuration for the pipeline steps.
         substitutions: Extra placeholders to use in the name templates.
+        cache_policy: Cache policy for this pipeline.
 
     Returns:
         A pipeline instance.
@@ -125,6 +129,7 @@ def pipeline(
             model=model,
             retry=retry,
             substitutions=substitutions,
+            cache_policy=cache_policy,
         )
 
         p.__doc__ = func.__doc__

zenml/pipelines/pipeline_definition.py

@@ -104,6 +104,7 @@ if TYPE_CHECKING:
     from zenml.artifacts.external_artifact import ExternalArtifact
     from zenml.client_lazy_loader import ClientLazyLoader
     from zenml.config.base_settings import SettingsOrDict
+    from zenml.config.cache_policy import CachePolicyOrString
     from zenml.config.retry_config import StepRetryConfig
     from zenml.config.source import Source
     from zenml.model.lazy_load import ModelVersionDataLazyLoader
@@ -145,6 +146,7 @@ class Pipeline:
         model: Optional["Model"] = None,
         retry: Optional["StepRetryConfig"] = None,
         substitutions: Optional[Dict[str, str]] = None,
+        cache_policy: Optional["CachePolicyOrString"] = None,
     ) -> None:
         """Initializes a pipeline.
 
@@ -170,6 +172,7 @@ class Pipeline:
             model: configuration of the model in the Model Control Plane.
             retry: Retry configuration for the pipeline steps.
             substitutions: Extra placeholders to use in the name templates.
+            cache_policy: Cache policy for this pipeline.
         """
         self._invocations: Dict[str, StepInvocation] = {}
         self._run_args: Dict[str, Any] = {}
@@ -193,6 +196,7 @@ class Pipeline:
                 model=model,
                 retry=retry,
                 substitutions=substitutions,
+                cache_policy=cache_policy,
             )
         self.entrypoint = entrypoint
         self._parameters: Dict[str, Any] = {}
@@ -316,6 +320,7 @@ class Pipeline:
         parameters: Optional[Dict[str, Any]] = None,
         merge: bool = True,
         substitutions: Optional[Dict[str, str]] = None,
+        cache_policy: Optional["CachePolicyOrString"] = None,
     ) -> Self:
         """Configures the pipeline.
 
@@ -355,6 +360,7 @@ class Pipeline:
             retry: Retry configuration for the pipeline steps.
             parameters: input parameters for the pipeline.
             substitutions: Extra placeholders to use in the name templates.
+            cache_policy: Cache policy for this pipeline.
 
         Returns:
             The pipeline instance that this method was called on.
@@ -390,6 +396,7 @@ class Pipeline:
                 "retry": retry,
                 "parameters": parameters,
                 "substitutions": substitutions,
+                "cache_policy": cache_policy,
             }
         )
         if not self.__suppress_warnings_flag__:

zenml/steps/base_step.py

@@ -35,6 +35,7 @@ from typing import (
 from pydantic import BaseModel, ConfigDict, ValidationError
 
 from zenml.client_lazy_loader import ClientLazyLoader
+from zenml.config.cache_policy import CachePolicyOrString
 from zenml.config.retry_config import StepRetryConfig
 from zenml.config.source import Source
 from zenml.constants import (
@@ -117,6 +118,7 @@ class BaseStep:
         model: Optional["Model"] = None,
         retry: Optional[StepRetryConfig] = None,
         substitutions: Optional[Dict[str, str]] = None,
+        cache_policy: Optional[CachePolicyOrString] = None,
     ) -> None:
         """Initializes a step.
 
@@ -146,6 +148,7 @@ class BaseStep:
             model: configuration of the model version in the Model Control Plane.
             retry: Configuration for retrying the step in case of failure.
             substitutions: Extra placeholders to use in the name template.
+            cache_policy: Cache policy for this step.
         """
         from zenml.config.step_configurations import PartialStepConfiguration
 
@@ -207,6 +210,7 @@ class BaseStep:
             model=model,
             retry=retry,
             substitutions=substitutions,
+            cache_policy=cache_policy,
         )
 
         notebook_utils.try_to_save_notebook_cell_code(self.source_object)
@@ -616,6 +620,7 @@ class BaseStep:
         model: Optional["Model"] = None,
         retry: Optional[StepRetryConfig] = None,
         substitutions: Optional[Dict[str, str]] = None,
+        cache_policy: Optional[CachePolicyOrString] = None,
         merge: bool = True,
     ) -> T:
         """Configures the step.
@@ -655,6 +660,7 @@ class BaseStep:
             model: Model to use for this step.
             retry: Configuration for retrying the step in case of failure.
             substitutions: Extra placeholders to use in the name template.
+            cache_policy: Cache policy for this step.
             merge: If `True`, will merge the given dictionary configurations
                 like `parameters` and `settings` with existing
                 configurations. If `False` the given configurations will
@@ -725,6 +731,7 @@ class BaseStep:
                 "model": model,
                 "retry": retry,
                 "substitutions": substitutions,
+                "cache_policy": cache_policy,
             }
         )
         config = StepConfigurationUpdate(**values)
@@ -750,6 +757,7 @@ class BaseStep:
         model: Optional["Model"] = None,
         retry: Optional[StepRetryConfig] = None,
         substitutions: Optional[Dict[str, str]] = None,
+        cache_policy: Optional[CachePolicyOrString] = None,
         merge: bool = True,
     ) -> "BaseStep":
         """Copies the step and applies the given configurations.
@@ -779,6 +787,7 @@ class BaseStep:
             model: Model to use for this step.
             retry: Configuration for retrying the step in case of failure.
             substitutions: Extra placeholders for the step name.
+            cache_policy: Cache policy for this step.
             merge: If `True`, will merge the given dictionary configurations
                 like `parameters` and `settings` with existing
                 configurations. If `False` the given configurations will
@@ -805,6 +814,7 @@ class BaseStep:
             model=model,
             retry=retry,
             substitutions=substitutions,
+            cache_policy=cache_policy,
             merge=merge,
         )
         return step_copy

zenml/steps/step_decorator.py

@@ -31,6 +31,7 @@ from zenml.logger import get_logger
 
 if TYPE_CHECKING:
     from zenml.config.base_settings import SettingsOrDict
+    from zenml.config.cache_policy import CachePolicyOrString
     from zenml.config.retry_config import StepRetryConfig
     from zenml.config.source import Source
     from zenml.materializers.base_materializer import BaseMaterializer
@@ -74,6 +75,7 @@ def step(
     model: Optional["Model"] = None,
     retry: Optional["StepRetryConfig"] = None,
     substitutions: Optional[Dict[str, str]] = None,
+    cache_policy: Optional["CachePolicyOrString"] = None,
 ) -> Callable[["F"], "BaseStep"]: ...
 
 
@@ -95,6 +97,7 @@ def step(
     model: Optional["Model"] = None,
     retry: Optional["StepRetryConfig"] = None,
     substitutions: Optional[Dict[str, str]] = None,
+    cache_policy: Optional["CachePolicyOrString"] = None,
 ) -> Union["BaseStep", Callable[["F"], "BaseStep"]]:
     """Decorator to create a ZenML step.
 
@@ -127,6 +130,7 @@ def step(
         model: configuration of the model in the Model Control Plane.
         retry: configuration of step retry in case of step failure.
         substitutions: Extra placeholders for the step name.
+        cache_policy: Cache policy for this step.
 
     Returns:
         The step instance.
@@ -161,6 +165,7 @@ def step(
             model=model,
             retry=retry,
             substitutions=substitutions,
+            cache_policy=cache_policy,
         )
 
         return step_instance

zenml/zen_stores/migrations/versions/aae4eed923b5_add_content_hash_column.py

@@ -0,0 +1,36 @@
+"""Add content hash column [aae4eed923b5].
+
+Revision ID: aae4eed923b5
+Revises: 0.84.3
+Create Date: 2025-08-22 16:32:12.494008
+
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "aae4eed923b5"
+down_revision = "0.84.3"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("artifact_version", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("content_hash", sa.TEXT(), nullable=True)
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("artifact_version", schema=None) as batch_op:
+        batch_op.drop_column("content_hash")
+
+    # ### end Alembic commands ###

zenml/zen_stores/schemas/artifact_schemas.py

@@ -299,6 +299,7 @@ class ArtifactVersionSchema(BaseSchema, RunMetadataInterface, table=True):
         ),
     )
     save_type: str = Field(sa_column=Column(TEXT, nullable=False))
+    content_hash: Optional[str] = Field(sa_column=Column(TEXT, nullable=True))
 
     # Foreign keys
     artifact_id: UUID = build_foreign_key_field(
@@ -483,6 +484,7 @@ class ArtifactVersionSchema(BaseSchema, RunMetadataInterface, table=True):
             materializer=artifact_version_request.materializer.model_dump_json(),
             data_type=artifact_version_request.data_type.model_dump_json(),
             save_type=artifact_version_request.save_type.value,
+            content_hash=artifact_version_request.content_hash,
         )
 
     def to_model(
@@ -530,6 +532,7 @@ class ArtifactVersionSchema(BaseSchema, RunMetadataInterface, table=True):
             updated=self.updated,
             save_type=ArtifactSaveType(self.save_type),
             artifact_store_id=self.artifact_store_id,
+            content_hash=self.content_hash,
         )
 
         # Create the metadata of the model

zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -599,6 +599,9 @@ class PipelineRunSchema(NamedSchema, RunMetadataInterface, table=True):
             self.status = run_update.status.value
             self.end_time = run_update.end_time
 
+        if run_update.orchestrator_run_id:
+            self.orchestrator_run_id = run_update.orchestrator_run_id
+
         self.updated = utc_now()
         return self
 

zenml/zen_stores/schemas/step_run_schemas.py

@@ -319,7 +319,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
             source_code=request.source_code,
             version=version,
             is_retriable=is_retriable,
-            exception_info=json.dumps(request.exception_info)
+            exception_info=request.exception_info.model_dump_json()
             if request.exception_info
             else None,
         )

zenml/zen_stores/sql_zen_store.py

@@ -5401,10 +5401,15 @@ class SqlZenStore(BaseZenStore):
                     step_id = step_run.id
                     metadata["status"] = step_run.status
 
-                    if step_run.end_time and step_run.start_time:
-                        metadata["duration"] = (
-                            step_run.end_time - step_run.start_time
-                        ).total_seconds()
+                    if step_run.start_time:
+                        metadata["start_time"] = (
+                            step_run.start_time.isoformat()
+                        )
+
+                        if step_run.end_time:
+                            metadata["duration"] = (
+                                step_run.end_time - step_run.start_time
+                            ).total_seconds()
 
                 step_node = helper.add_step_node(
                     node_id=helper.get_step_node_id(name=step_name),
@@ -6142,6 +6147,9 @@ class SqlZenStore(BaseZenStore):
         Raises:
             EntityExistsError: If a log entry with the same source already
                 exists within the scope of the same pipeline run.
+            IllegalOperationError: If the orchestrator run id is being updated
+                on a non-placeholder run or if the orchestrator run id is
+                already set and is different from the new orchestrator run id.
         """
         with Session(self.engine) as session:
             # Check if pipeline run with the given ID exists
@@ -6151,6 +6159,24 @@ class SqlZenStore(BaseZenStore):
                 session=session,
             )
 
+            if run_update.orchestrator_run_id:
+                if not existing_run.is_placeholder_run():
+                    raise IllegalOperationError(
+                        "Cannot update the orchestrator run id of a pipeline "
+                        "run that is not a placeholder run."
+                    )
+
+                if (
+                    existing_run.orchestrator_run_id
+                    and existing_run.orchestrator_run_id
+                    != run_update.orchestrator_run_id
+                ):
+                    raise IllegalOperationError(
+                        "Pipeline run already has a different orchestrator run "
+                        f"id. Existing: {existing_run.orchestrator_run_id}, "
+                        f"New: {run_update.orchestrator_run_id}"
+                    )
+
             existing_run.update(run_update=run_update)
             session.add(existing_run)
             session.commit()

{zenml_nightly-0.84.3.dev20250903.dist-info → zenml_nightly-0.84.3.dev20250906.dist-info}/METADATA

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

{zenml_nightly-0.84.3.dev20250903.dist-info → zenml_nightly-0.84.3.dev20250906.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=ouJDJKYOvUFMlyUN-y3sdOniHgu3Q8Bo4ArSugmDIOk,19
+zenml/VERSION,sha256=CER6yVDu4ppYm9l-x_S_uCuRfUAdGMCdCzvVQ1p3M1E,19
 zenml/__init__.py,sha256=r7JUg2SVDf_dPhS7iU6vudKusEqK4ics7_jFMZhq0o4,2731
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -25,7 +25,7 @@ zenml/artifacts/external_artifact.py,sha256=7nLANV0vsGC36H1s_B_awX4hnZgXHCGIscQ2
 zenml/artifacts/external_artifact_config.py,sha256=P172p0JOu8Xx1F8RCQut1dnRpt5lpWXGNqMbY-V90sI,3323
 zenml/artifacts/preexisting_data_materializer.py,sha256=dcahDcHUD3Lvn0-6zE2BG84bkyo_ydAgzBWxtbyJJZQ,3325
 zenml/artifacts/unmaterialized_artifact.py,sha256=JNPKq_sNifQx5wP8jEw7TGBEi26zwKirPGlWX9uxbJI,1300
-zenml/artifacts/utils.py,sha256=U2owJEOxOqRHG4dJ3y2HiPxUUWkaIxHVVmR5_iLn1Ow,36073
+zenml/artifacts/utils.py,sha256=5O8YSHAyoXenCNprnramrv4urb_PdlL6gyO_HNwae_Q,36168
 zenml/cli/__init__.py,sha256=Q-DTKGMxUCyN8aAchAvjoWJeZ3f55Hnt4FpagiNHSbU,75655
 zenml/cli/annotator.py,sha256=JRR7_TJOWKyiKGv1kwSjG1Ay6RBWPVgm0X-D0uSBlyE,6976
 zenml/cli/artifact.py,sha256=7lsAS52DroBTFkFWxkyb-lIDOGP5jPL_Se_RDG_2jgg,9564
@@ -62,15 +62,16 @@ zenml/code_repositories/base_code_repository.py,sha256=Id6VjbUu8N3ZpNvBGhOgbahto
 zenml/code_repositories/git/__init__.py,sha256=vU8UzMp8sv9n-R2r7VKa9LdQcYER6BhO4O-z8Ppa3kM,824
 zenml/code_repositories/git/local_git_repository_context.py,sha256=PuD4GOnxBNtkSddc5QwMnnx3F2t4gClUeGmhfoLQHfs,5780
 zenml/code_repositories/local_repository_context.py,sha256=1VyiYkJBDVg0iGusgRQDToGRPJuu9lx7jTBDpplukDg,2816
-zenml/config/__init__.py,sha256=NIMS7QQo3-uVkNlzPvR5FbtgPdRibF2xFJB7Vr0fhCM,1826
+zenml/config/__init__.py,sha256=KRAcMVUgE5R5cvoVQjs1lHUcl-IYBKmDSeBnwRIPG04,1265
 zenml/config/base_settings.py,sha256=itoLqc1cOwEYhgSGdZmSKSaBevQkvYH7NQh7PUamazc,1700
 zenml/config/build_configuration.py,sha256=jGGNwP0Cb7a80JXArNxDgxzxl9ytSZRtv-WW7_psLbM,6870
-zenml/config/compiler.py,sha256=uezFp9RKBhbh-zZSYMJ-QJNZQ4ksitsBsHX8mvR11jI,23802
+zenml/config/cache_policy.py,sha256=_bXEbR7s8xt78JO-BUJ1V2clI3lO7FNMF3r6YbD_s8M,3121
+zenml/config/compiler.py,sha256=YY1C-y_NBn0RpVCHfde0TQ01CJXLn01VOsytPpowEWg,23852
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=xZvoeC6v6RG_5MFK_RgdTgrkOLu-QT6frY4SGQNZUTo,18460
 zenml/config/global_config.py,sha256=59orl6xp4moxSo2Q8tXtDfpIBX4AxJ9FHcm8s2VHyWw,29712
-zenml/config/pipeline_configurations.py,sha256=7trCbElpqGGgawii2FrdLW8fKaAWCR8jACkNqdG_vcQ,3983
-zenml/config/pipeline_run_configuration.py,sha256=Y9C5mVUH-w4gc1w1PCQFpjEmfBBpSMvb8riA_sL78hY,2311
+zenml/config/pipeline_configurations.py,sha256=3AVe-VYwFkccpdpb4aH8w2JZCwD_Bh4V25f4YAufbwE,4106
+zenml/config/pipeline_run_configuration.py,sha256=zZQS10-3qEnjXE9yZ7dJnnfYKh6mX-8Nwybjag-Ywpk,2434
 zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE,2821
 zenml/config/resource_settings.py,sha256=0taXGHvDfXuvpMplxsuzpznB1sAvWJIGnXoVJ9-ySfw,3899
 zenml/config/retry_config.py,sha256=rld2nvjGpX67Wci6abCA5T7vAi-TVgM30gT5nT8N_8M,1009
@@ -80,7 +81,7 @@ zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wz
 zenml/config/server_config.py,sha256=HNHODfojAstBWMxcV5OjxcXSh8Oy9k-_0J5la60fqzQ,32616
 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=TFRYwzCnwWY69m1Cg9jmAl8eTUhD7wSgS05FoaMCnXg,14817
+zenml/config/step_configurations.py,sha256=KVjcAAtNmEcDkEyLjW1tVPHEHBo9dnu25dCoj9JV6rg,15088
 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
@@ -129,7 +130,7 @@ zenml/integrations/airflow/flavors/__init__.py,sha256=Y48mn5OxERPPaXDBd5CFAIn6yh
 zenml/integrations/airflow/flavors/airflow_orchestrator_flavor.py,sha256=VfZQD2H-WwIgVD1Fi7uewdnkvRoSykY0YCfROFDadXg,6189
 zenml/integrations/airflow/orchestrators/__init__.py,sha256=UPly5wqlFdzLSiCeEtIExRXjTNdO0ys2tHSEbEv1Dks,845
 zenml/integrations/airflow/orchestrators/airflow_orchestrator.py,sha256=2o1aJlUPjDH-jI1Sfwha0-rW0FYaTyU2E7wEh9OTel4,15944
-zenml/integrations/airflow/orchestrators/dag_generator.py,sha256=H5OWZte6q1mXR-joXCbpKJ2KGPIhSOCL0ok-RIiZ1OQ,7345
+zenml/integrations/airflow/orchestrators/dag_generator.py,sha256=ewRPwB2yl-a25JrbPGxeu2yCUoOZRiDCtfq_AfEuqJs,7336
 zenml/integrations/argilla/__init__.py,sha256=LmGOS77UN3jODugqrqXR_6oAg1JC6HN8srY1VgmfpgM,1415
 zenml/integrations/argilla/annotators/__init__.py,sha256=QjRMxIQ-skulcLN94GuHuukrmZH27e8lke_tXUp8MOg,798
 zenml/integrations/argilla/annotators/argilla_annotator.py,sha256=aIA5rcwfOYLnUVp2c_aaTAUnVW8GV9gP8myze_jY_qY,15586
@@ -341,7 +342,7 @@ zenml/integrations/kubernetes/orchestrators/__init__.py,sha256=TJID3OTieZBox36Wp
 zenml/integrations/kubernetes/orchestrators/dag_runner.py,sha256=IDumuhi9hySiTZ7oTI6hz5RlJ4kiuv3NpVYgA_pLS_Y,12044
 zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=XXBs-iE77X0J6K-uRQxXK-iir4IY-Uz3p1b82iQlJ9k,37335
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=TnrKa6xnrYtAq1elyob0aRy-aTLrSjLuuOzdLSs_GFs,36811
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=QahXJ-cM6lrXqWs2BHMQtKQf3svqKB5l69d4D0OB4qc,24789
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=bDZammsOjls-81jaLjC5J0JJy_49eQ0MxXgqCAnVHaY,24992
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=QOwQnWCfB-t_BQ2eOZN0SakurGUd0GTMCSdUlREhk6I,2324
 zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=JskjDSbR3HdTxGT3NOVi0ha7Ku3CqMkPzMJJq4IZYp0,14491
 zenml/integrations/kubernetes/pod_settings.py,sha256=s6I0sALXB7PnRqJ-nlU71o3cTm2GM1TfqsE44qhiDJY,8022
@@ -598,17 +599,17 @@ zenml/login/pro/workspace/models.py,sha256=godIY1q2dUxB0QILaOFOeqV1I3WglWUjZC11n
 zenml/login/server_info.py,sha256=-_sK2L-curHdzUv1JDOwF6GoEeAXT5vFZN0J-5Ug4wU,1663
 zenml/login/web_login.py,sha256=aAwqi4-AD5dJVJ3pCGi6aCIaa-dAFXOqnTdqRlwh-CU,9417
 zenml/materializers/__init__.py,sha256=C3lZaTmIFxwIPwCKF8oLQUkLaX2o7Dbj9hvYVFrSzt8,1758
-zenml/materializers/base_materializer.py,sha256=wMRLOAz0arRU2Q6tCxcYVfT62I-qg7HVHOPUd1OtKAg,13939
-zenml/materializers/built_in_materializer.py,sha256=HgcCHg3GpuQ4t-jecYcdg0kldUfLeUJvIpm8-wcS11I,17189
+zenml/materializers/base_materializer.py,sha256=L77R7YTbFW_P_bHcbSSDNUmY4WxaqJtaZ8QVmSGTVN0,14231
+zenml/materializers/built_in_materializer.py,sha256=xExGEG8f1T4CeTkjevIsG0exYy4ReNWesBz9xEVXe1o,19060
 zenml/materializers/cloudpickle_materializer.py,sha256=PIZauXg1SBYmBUi9FBd2t3LFyZupp3jYtMjZ0OTug18,4816
 zenml/materializers/materializer_registry.py,sha256=ic-aWhJ2Ex9F_rml2dDVAxhRfW3nd71QMxzfTPP6BIM,4002
 zenml/materializers/numpy_materializer.py,sha256=OLcHF9Z0tAqQ_U8TraA0vGmZjHoT7eT_XevncIutt0M,1715
 zenml/materializers/pandas_materializer.py,sha256=c4B-ly04504gysA66iCYcmEdeh0ClePRTxRCkmHqIgE,1725
 zenml/materializers/path_materializer.py,sha256=ydfKHhi14acjSb9jqizsXRRO4wKaCsSNUJg2woPw9XY,6217
-zenml/materializers/pydantic_materializer.py,sha256=eDp3eOR-X7FOHlpwALOJtVUtJti75Dsa7r0lzSjeQ-Q,2271
+zenml/materializers/pydantic_materializer.py,sha256=PCY399H9L7i3dxHCLl2QVypoQAG3h2eAZZyZ0QZctD4,2865
 zenml/materializers/service_materializer.py,sha256=OV5HFUuIc8UF8vE5y_FTWn4_mpCz0AtJPnNA2YFAjZE,2992
-zenml/materializers/structured_string_materializer.py,sha256=8LIvSH3JQn_QRAWGXK5_NhPQ4NgCt5zzz6f8GnGG9q4,4442
-zenml/materializers/uuid_materializer.py,sha256=2lrTDZlq5TAE1EhT-S7GPP03CzrTDTzVhb7UScz9zgs,2629
+zenml/materializers/structured_string_materializer.py,sha256=2rhGzoRfulwrjXV2b9WNrjoJZBk7Nibq2C47xVxuIpM,4929
+zenml/materializers/uuid_materializer.py,sha256=YTJeiSTJeqd2TXuFT4IKLpncnWVHRPPu0zeh4E5Tx_k,3094
 zenml/metadata/__init__.py,sha256=iklF7cvq9BqHiYOC38qIm-LeE2I29Jl-RVl33hpS5Ao,778
 zenml/metadata/lazy_load.py,sha256=DtYSkhdTr2LDJoq2GYkwfLHy-3xcVoOFu_y4PKhVU_o,2903
 zenml/metadata/metadata_types.py,sha256=ts1EhF2qGZb8siKv1nkPSeFeyR2kbiIODkpk-hyoTxE,6745
@@ -634,7 +635,7 @@ zenml/models/v2/core/action_flavor.py,sha256=DlGlJTOcgLSDYAsrbAyXOJVdo7X2GYTFQTT
 zenml/models/v2/core/api_key.py,sha256=0iye9ftNk-Gjg4izZqYqbNgF1UR51WdRfasiAE2TqPw,11536
 zenml/models/v2/core/api_transaction.py,sha256=NepWmgi3q_1B5qiKho2nBWxAPGRVNaUzjgt-Rj5ZDCU,5104
 zenml/models/v2/core/artifact.py,sha256=1b0VZPEez3mi6pknfN5w10FgsxJeStyOgv4zjCM8uSs,8177
-zenml/models/v2/core/artifact_version.py,sha256=v15ZlJYZnheqvgeYN6VqPQlbFyxEBKoGfvW-Eb0k50Y,23577
+zenml/models/v2/core/artifact_version.py,sha256=HWQuQqkR2qm4TzZ0GgqKJYLmqePLKbVN94gjFUXPYUY,24087
 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
@@ -651,7 +652,7 @@ zenml/models/v2/core/model_version_pipeline_run.py,sha256=JbPZZEQvOK9I32htkWdAON
 zenml/models/v2/core/pipeline.py,sha256=OXArih3YodMAZBS_GzuLL6VPpFv3c59EthwDfDZiNGk,12044
 zenml/models/v2/core/pipeline_build.py,sha256=z0LCc8aR8AdNFLLtzaAP0U0ffv7WpmSx9nNAyI__e14,17008
 zenml/models/v2/core/pipeline_deployment.py,sha256=nrJHrweNlwFsIdExINMF7jwhjOwBpOd8iwtEyAnvlz4,11971
-zenml/models/v2/core/pipeline_run.py,sha256=lGMzt2AS9-5kHFZa6MBDfx68LLJu4ohMRGhtelwfYHA,31016
+zenml/models/v2/core/pipeline_run.py,sha256=LJRNc2tR42SWiyyN5njcdAh-O5RJo8ihboW-ATfEGS0,31062
 zenml/models/v2/core/project.py,sha256=fNNO8Tg5OhSzmFf2t6g4SpUzGWC96oHhUccVyWytvIE,5627
 zenml/models/v2/core/run_metadata.py,sha256=hRGQa_sk99uDSab3EyyOQhefypVpiQDCH3oAtblREDk,2432
 zenml/models/v2/core/run_template.py,sha256=6jdH1son7kpvFv-vtaOL2nXMATtVpqk_7a2xOVv_7cc,14097
@@ -685,8 +686,8 @@ zenml/models/v2/misc/statistics.py,sha256=ajce9rHC2bBylLzOmLfcOBSbTnJD67Y8n5YKCY
 zenml/models/v2/misc/tag.py,sha256=jUoz7wqMpDFlIUmvj4PVq8NYJJB7TMCcdRfW-DAABCU,974
 zenml/models/v2/misc/user_auth.py,sha256=1-yafNA9qK4wL8ToROjaklTVI7Mj9va0t80_4wm7w3U,6988
 zenml/orchestrators/__init__.py,sha256=Nhmc6U-q6c8TEH1Jb5l8XaKnc4KmLNspDpvvV5TcvzQ,2007
-zenml/orchestrators/base_orchestrator.py,sha256=m9qsavtb4fQv1pie8cnuQlOVxu_p2qJQ8X7jukijFZ4,21445
-zenml/orchestrators/cache_utils.py,sha256=QkmTs-ANfXve9_QzTqgGlyulZDEWOngoTcsiSjG5aA8,5906
+zenml/orchestrators/base_orchestrator.py,sha256=-WbVagJ8PEs0-RVAa_mSFFUQqn0L4npjwBwN16LQ_5Y,21760
+zenml/orchestrators/cache_utils.py,sha256=PLZwNZREx3W4ZgRUahuIbfeTdR08uNWcwuDeoxzANac,6537
 zenml/orchestrators/containerized_orchestrator.py,sha256=-umxlNlQfezks1GLoi0SyXtXXo9dkC4Re9nA0fh_avw,3858
 zenml/orchestrators/dag_runner.py,sha256=6pDloHKuGvOTaABJ8G8OJ94f_8uPfeN-M0Nc5sFjHwA,10708
 zenml/orchestrators/input_utils.py,sha256=IwDYEwhFVcvZcdi9Myg__UB1bkZfpa8xZ0XRSwc4-go,6934
@@ -697,7 +698,7 @@ zenml/orchestrators/local_docker/local_docker_orchestrator.py,sha256=dEc1R2_xzFI
 zenml/orchestrators/output_utils.py,sha256=01vqke1ZfmfuLpgxNerF-QL2wA0VPv1zUdvlMw0OwUY,3508
 zenml/orchestrators/publish_utils.py,sha256=u0jk9Q57fLQE1QDJjrydy5KQ9xtYKUvn3CXBirDH2u8,8485
 zenml/orchestrators/step_launcher.py,sha256=dOvvMtgO_UhzPJYR3cz2H_j4L74qd9r9ggIc7EYzj40,18767
-zenml/orchestrators/step_run_utils.py,sha256=JXlYomV_PmC-cmjk7VWAGauKUXdijXU0ao8HBTZobmY,16802
+zenml/orchestrators/step_run_utils.py,sha256=52ANscGnPTAyrOcMfMzSb3s3udxLiDL3Ly1uDvd2ZRk,16655
 zenml/orchestrators/step_runner.py,sha256=X3sX7f5NBYxT3utFGKcOJlkvCauYyEoo0LB2TB2Pp9U,27435
 zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7cM,5784
 zenml/orchestrators/utils.py,sha256=6bqLc1fmdJTXg8JUwUKs8YNbmxTuMIfWmUbUpg-7hx0,12956
@@ -705,8 +706,8 @@ zenml/orchestrators/wheeled_orchestrator.py,sha256=eOnMcnd3sCzfhA2l6qRAzF0rOXzao
 zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,873
 zenml/pipelines/build_utils.py,sha256=n7PyBGKPYGGc-QLuf4IfZE7iqlFMu2S2dnpI4TGAN6c,27952
 zenml/pipelines/pipeline_context.py,sha256=4BixReLcPo33VtNBDrMwnJqjKTinHjmO5AOfmoeIOQM,3659
-zenml/pipelines/pipeline_decorator.py,sha256=vR6prQPyxJ7lvSmr3sfb3lJ7LjEv3fx8NftmmNH9EUc,4836
-zenml/pipelines/pipeline_definition.py,sha256=v8wwZER4dGKztn_B2HLCXMXma1SAYydHvuV3Z480y04,60143
+zenml/pipelines/pipeline_decorator.py,sha256=4UO9JZuO7PylwqEmcYpCTcWND9bFBdaMS7dK5mbd4gw,5107
+zenml/pipelines/pipeline_definition.py,sha256=UeGemwJ_TMWlVCdOnm5g_b8Vk55YSvFhmQc8Uo6WOuA,60534
 zenml/pipelines/run_utils.py,sha256=VAjfdu300AKfTuwXll3uoFrwN5dt_hesXxtylndUraQ,12515
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
@@ -756,11 +757,11 @@ zenml/step_operators/__init__.py,sha256=tqj7fgnQyZubLjwUu4ITwkA-70KMQz4g37agbfF7
 zenml/step_operators/base_step_operator.py,sha256=ZRnY6lcEHY8xZskrKKdPhgKg2BlEoh2_kb8xCaM2D1g,3522
 zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=wuQKmEI_ckn1CHHrxWGGdIhQyBFXG01aKC2-2b6Atjo,3690
 zenml/steps/__init__.py,sha256=KKWFOmCZGLDEikOD2E5YmDA7QHo47uPV37by21WwI0U,1453
-zenml/steps/base_step.py,sha256=HUNNQe3A5DyzEHm4gAbX_NCC2LiKRDI6nrngURW6WdI,44632
+zenml/steps/base_step.py,sha256=6XncnFSONWfezBcWzatevo-GPRJ2P5uVuigCIY9iEIs,45156
 zenml/steps/decorated_step.py,sha256=cr4m7qi525fUc_VTkb7J16dKdUSzc4UA5eyHg8vFWQg,4275
 zenml/steps/entrypoint_function_utils.py,sha256=H0WIkHpR_R0S9gl_tWr0nX-fcBlYnm8OQ1cimvrw-qo,9555
 zenml/steps/step_context.py,sha256=sFvVVvEyKmWTrucofor8Cuxv-72jbziVaQY-OlOvFAM,15526
-zenml/steps/step_decorator.py,sha256=0TKxP_oAVRizjN3JhBZ8ShFd3fSBdVPU2ijytTg1h2U,6506
+zenml/steps/step_decorator.py,sha256=DDuxBCDLAVjVmW3d3_ASUuKV9zK7CZkUCBwg1l2M2oY,6773
 zenml/steps/step_invocation.py,sha256=ETfOaV-P4_iXGk9y1-xK54Kfg2QRYaGoj_jTyEYZfb0,4861
 zenml/steps/utils.py,sha256=UpZ5lN46Bqu1eXyVfh969M4-UBbOZjq5R7u4PkXYDZ8,18586
 zenml/types.py,sha256=LhGWJ4t3nybBk1l9Ox3tqqHbTYSuCqhkRsL5FqO6yf4,1206
@@ -1279,6 +1280,7 @@ zenml/zen_stores/migrations/versions/__init__.py,sha256=ab-drj6BQO7OuDMGvcSlAjxG
 zenml/zen_stores/migrations/versions/a1237ba94fd8_add_model_version_producer_run_unique_.py,sha256=6H-TsiZWbzvhW8fvVpta3wzpK7Shc7RdHV-6At3IczI,2126
 zenml/zen_stores/migrations/versions/a39c4184c8ce_remove_secrets_manager_flavors.py,sha256=9bs6F1GSUu_kerajVB9nj_zIkcEa1NmLJ5N2quNFd9Q,794
 zenml/zen_stores/migrations/versions/a91762e6be36_artifact_version_table.py,sha256=EmViRqR-vcm2lZU6itSEKQzntJtg1FdzrE2Mmgozn6c,7818
+zenml/zen_stores/migrations/versions/aae4eed923b5_add_content_hash_column.py,sha256=r-Ewuzjb5fxftnyVkd35nmMd-LKNlUQGT-F6X3ohhz8,1018
 zenml/zen_stores/migrations/versions/ade72effebaf_added_logs_table.py,sha256=BP-WuS8xSM0fUARtTxzge3g9AyznNPU7OX3N3tgx6Sw,1963
 zenml/zen_stores/migrations/versions/alembic_start.py,sha256=707q8fI4cwfLzBqVcVqqt5aF_H9i5eI5AHWkjrt_sb0,15702
 zenml/zen_stores/migrations/versions/b4eccf34dfa3_add_hub_token_to_user_model.py,sha256=j9ifD7KC18pgzyCIL6tDa4rRpWi7uDY97pFvV-abTpc,1075
@@ -1314,7 +1316,7 @@ zenml/zen_stores/schemas/__init__.py,sha256=myNtX0C4sr3spHJi7fyQWY8DGOD1UVrUWzzI
 zenml/zen_stores/schemas/action_schemas.py,sha256=jymEro5r20seTpRcAr5iZw-g8rXZRSZFEbiEm_iFaj8,7618
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=RJS-lH7AHHAe9oXA0CGWvtbrmYgVfNP9nMa7rhtRNnY,8379
 zenml/zen_stores/schemas/api_transaction_schemas.py,sha256=3shikZ5Vb9xvknEIAfwz4udcfEdUSbSwQN6E9WHLcNU,4359
-zenml/zen_stores/schemas/artifact_schemas.py,sha256=P4k8w8Baj7VEM5cPBLFMUBS4dvinZ8RBOEhX8YDH7D0,19822
+zenml/zen_stores/schemas/artifact_schemas.py,sha256=626uxbBvQ4JGx1KW9YnXkYpd43d_HenKmfAGo6kK524,20009
 zenml/zen_stores/schemas/artifact_visualization_schemas.py,sha256=zv_Mg--WMTtAPEy5PXIkZlxq-oLrXZNLHXb_YgZbm0g,3667
 zenml/zen_stores/schemas/base_schemas.py,sha256=pS5aEeXgcOI5PENmY_FzfDCH-lCqPiVd3jGd-0iD2t8,2906
 zenml/zen_stores/schemas/code_repository_schemas.py,sha256=rC7PUoG9MvnnmtaQdm5JcOdpcEt7yqEDVzEn17v7uI8,8858
@@ -1327,7 +1329,7 @@ zenml/zen_stores/schemas/logs_schemas.py,sha256=la8ipJeTZ920ubHn8Uqf1DJLR7xHEXKK
 zenml/zen_stores/schemas/model_schemas.py,sha256=cDhWggrn3rTjaFML3iQGuW_4CpUJUxAnHieiY-dq0y4,26006
 zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=8GMdJNNcoSnbYH9daVr8zrhwQ1n-HZrpgzHoBZ7DZyA,7320
 zenml/zen_stores/schemas/pipeline_deployment_schemas.py,sha256=ZJ5K_rzVhe7rvE6kRdmYhRn4E2DcZxxMUPgAImeLN2s,15423
-zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=EO1AtuHuWznSdWb5lh87zG1g1l49B5Xg1MyIGHaHSps,24140
+zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=yTXvL72X_v2ymJxcP1lAF0x0BkxSvvG9Oja6qI-RJqg,24254
 zenml/zen_stores/schemas/pipeline_schemas.py,sha256=xgioTeBuFFFDOJi5eESx2j-8mW55B6hshosFylev5Mw,8213
 zenml/zen_stores/schemas/project_schemas.py,sha256=X2GClPNQz0COsEZX8xI-I8Sm68Hb6f20Obm24mQyLS0,6013
 zenml/zen_stores/schemas/run_metadata_schemas.py,sha256=G94rT4ldluMSnf9rm7R_9rw_GlgaAyq72ptkHl0gHeg,3605
@@ -1339,7 +1341,7 @@ zenml/zen_stores/schemas/server_settings_schemas.py,sha256=usBE-idRrmK-LeLN0zDtC
 zenml/zen_stores/schemas/service_connector_schemas.py,sha256=SpzFZTbfxypFtMLBujPSlGOpt-kzjcHfq3IoBghPizI,11137
 zenml/zen_stores/schemas/service_schemas.py,sha256=Fn_DyGm-PF-qN1TPZA3Q8AhrnUOTSIX0V_MD0laZ2Bk,10685
 zenml/zen_stores/schemas/stack_schemas.py,sha256=DOPoNwblTqN83RpnD6cjyF80VeplA1BGUX8yuulpzKk,6829
-zenml/zen_stores/schemas/step_run_schemas.py,sha256=lzisqMKUsKmzfl6AiagFlmVADOZowFvyZamltgjXxaE,20740
+zenml/zen_stores/schemas/step_run_schemas.py,sha256=WIPhUfs7ez2-mTsvArnThoFnlZZwN5D1uelS-PEp9Ks,20746
 zenml/zen_stores/schemas/tag_schemas.py,sha256=pwQ5ogZovtkUuRtAkHakOfMh6ixw6qofvxB4i_Gd4qA,8352
 zenml/zen_stores/schemas/trigger_schemas.py,sha256=LokauFEHdN5UuYeC2kf5PtJ60WMz-E82Ngcwxcv6tDA,11936
 zenml/zen_stores/schemas/user_schemas.py,sha256=HmfwW_QALw4UdtWplfaX7Ls6LIoTo110y1VodADmOhg,12091
@@ -1353,11 +1355,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=5err1a-TrV3SR5
 zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=DrXGMkBxQIy2n_kkux5Xh2OM3Ks3MOpqP1D4aY8bfyY,7047
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
-zenml/zen_stores/sql_zen_store.py,sha256=nGuLAhFMlDUMaL9DwpwmuDCSZrJqyl60tIxdN6E3mH0,491850
+zenml/zen_stores/sql_zen_store.py,sha256=T40n0lPnai6UW16FZM7neBEjgcfz0VRnnkb20SKivDw,493073
 zenml/zen_stores/template_utils.py,sha256=iCXrXpqzVTY7roqop4Eh9J7DmLW6PQeILZexmw_l3b8,10074
 zenml/zen_stores/zen_store_interface.py,sha256=weiSULdI9AsbCE10a5TcwtybX-BJs9hKhjPJnTapWv4,93023
-zenml_nightly-0.84.3.dev20250903.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.84.3.dev20250903.dist-info/METADATA,sha256=s-rHNx_7rfqNAYEJo66y_V94H7RzKxgTfLvyeR9AmIo,25252
-zenml_nightly-0.84.3.dev20250903.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.84.3.dev20250903.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.84.3.dev20250903.dist-info/RECORD,,
+zenml_nightly-0.84.3.dev20250906.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.84.3.dev20250906.dist-info/METADATA,sha256=5tSzujncpH8sSECT2ij_qhv9CE-2cXDRiSCnlrI7pKw,25252
+zenml_nightly-0.84.3.dev20250906.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.84.3.dev20250906.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.84.3.dev20250906.dist-info/RECORD,,