zenml-nightly 0.73.0.dev20250205__py3-none-any.whl → 0.73.0.dev20250206__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.73.0.dev20250205
+0.73.0.dev20250206

zenml/cli/utils.py

@@ -18,6 +18,7 @@ import json
 import os
 import platform
 import re
+import shutil
 import subprocess
 import sys
 from typing import (
@@ -1048,22 +1049,18 @@ def install_packages(
         # just return without doing anything
         return
 
-    pip_command = ["uv", "pip"] if use_uv else ["pip"]
-    if upgrade:
-        command = (
-            [
-                sys.executable,
-                "-m",
-            ]
-            + pip_command
-            + [
-                "install",
-                "--upgrade",
-            ]
-            + packages
-        )
+    if use_uv and not is_installed_in_python_environment("uv"):
+        # If uv is installed globally, don't run as a python module
+        command = []
     else:
-        command = [sys.executable, "-m"] + pip_command + ["install"] + packages
+        command = [sys.executable, "-m"]
+
+    command += ["uv", "pip", "install"] if use_uv else ["pip", "install"]
+
+    if upgrade:
+        command += ["--upgrade"]
+
+    command += packages
 
     if not IS_DEBUG_ENV:
         quiet_flag = "-q" if use_uv else "-qqq"
@@ -1094,62 +1091,54 @@ def uninstall_package(package: str, use_uv: bool = False) -> None:
         package: The package to uninstall.
         use_uv: Whether to use uv for package uninstallation.
     """
-    pip_command = ["uv", "pip"] if use_uv else ["pip"]
-    quiet_flag = "-q" if use_uv else "-qqq"
-
-    if use_uv:
-        subprocess.check_call(
-            [
-                sys.executable,
-                "-m",
-            ]
-            + pip_command
-            + [
-                "uninstall",
-                quiet_flag,
-                package,
-            ]
-        )
+    if use_uv and not is_installed_in_python_environment("uv"):
+        # If uv is installed globally, don't run as a python module
+        command = []
     else:
-        subprocess.check_call(
-            [
-                sys.executable,
-                "-m",
-            ]
-            + pip_command
-            + [
-                "uninstall",
-                quiet_flag,
-                "-y",
-                package,
-            ]
-        )
+        command = [sys.executable, "-m"]
 
+    command += (
+        ["uv", "pip", "uninstall", "-q"]
+        if use_uv
+        else ["pip", "uninstall", "-y", "-qqq"]
+    )
+    command += [package]
 
-def is_uv_installed() -> bool:
-    """Check if uv is installed in the current environment.
+    subprocess.check_call(command)
+
+
+def is_installed_in_python_environment(package: str) -> bool:
+    """Check if a package is installed in the current python environment.
+
+    Args:
+        package: The package to check.
 
     Returns:
-        True if uv is installed, False otherwise.
+        True if the package is installed, False otherwise.
     """
     try:
-        pkg_resources.get_distribution("uv")
+        pkg_resources.get_distribution(package)
         return True
     except pkg_resources.DistributionNotFound:
         return False
 
 
+def is_uv_installed() -> bool:
+    """Check if uv is installed.
+
+    Returns:
+        True if uv is installed, False otherwise.
+    """
+    return shutil.which("uv") is not None
+
+
 def is_pip_installed() -> bool:
     """Check if pip is installed in the current environment.
 
     Returns:
         True if pip is installed, False otherwise.
     """
-    try:
-        pkg_resources.get_distribution("pip")
-        return True
-    except pkg_resources.DistributionNotFound:
-        return False
+    return is_installed_in_python_environment("pip")
 
 
 def pretty_print_secret(

zenml/client.py

@@ -3821,7 +3821,7 @@ class Client(metaclass=ClientMetaClass):
         templatable: Optional[bool] = None,
         tag: Optional[str] = None,
         user: Optional[Union[UUID, str]] = None,
-        run_metadata: Optional[Dict[str, str]] = None,
+        run_metadata: Optional[Dict[str, Any]] = None,
         pipeline: Optional[Union[UUID, str]] = None,
         code_repository: Optional[Union[UUID, str]] = None,
         model: Optional[Union[UUID, str]] = None,
@@ -3974,6 +3974,7 @@ class Client(metaclass=ClientMetaClass):
         user: Optional[Union[UUID, str]] = None,
         model_version_id: Optional[Union[str, UUID]] = None,
         model: Optional[Union[UUID, str]] = None,
+        run_metadata: Optional[Dict[str, Any]] = None,
         hydrate: bool = False,
     ) -> Page[StepRunResponse]:
         """List all pipelines.
@@ -4000,6 +4001,7 @@ class Client(metaclass=ClientMetaClass):
             cache_key: The cache key of the step run to filter by.
             code_hash: The code hash of the step run to filter by.
             status: The name of the run to filter by.
+            run_metadata: Filter by run metadata.
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
 
@@ -4028,6 +4030,7 @@ class Client(metaclass=ClientMetaClass):
             user=user,
             model_version_id=model_version_id,
             model=model,
+            run_metadata=run_metadata,
         )
         step_run_filter_model.set_scope_workspace(self.active_workspace.id)
         return self.zen_store.list_run_steps(
@@ -4254,7 +4257,7 @@ class Client(metaclass=ClientMetaClass):
         user: Optional[Union[UUID, str]] = None,
         model: Optional[Union[UUID, str]] = None,
         pipeline_run: Optional[Union[UUID, str]] = None,
-        run_metadata: Optional[Dict[str, str]] = None,
+        run_metadata: Optional[Dict[str, Any]] = None,
         tag: Optional[str] = None,
         hydrate: bool = False,
     ) -> Page[ArtifactVersionResponse]:
@@ -4320,6 +4323,7 @@ class Client(metaclass=ClientMetaClass):
             user=user,
             model=model,
             pipeline_run=pipeline_run,
+            run_metadata=run_metadata,
         )
         artifact_version_filter_model.set_scope_workspace(
             self.active_workspace.id

zenml/models/__init__.py

@@ -28,6 +28,7 @@ from zenml.models.v2.base.base import (
     BaseDatedResponseBody,
 )
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
     UserScopedRequest,
     UserScopedFilter,
     UserScopedResponse,
@@ -39,7 +40,7 @@ from zenml.models.v2.base.scoped import (
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
+    WorkspaceScopedFilter,
 )
 from zenml.models.v2.base.filter import (
     BaseFilter,
@@ -497,12 +498,13 @@ __all__ = [
     "WorkspaceScopedResponseBody",
     "WorkspaceScopedResponseMetadata",
     "WorkspaceScopedResponseResources",
-    "WorkspaceScopedTaggableFilter",
+    "WorkspaceScopedFilter",
     "BaseFilter",
     "StrFilter",
     "BoolFilter",
     "NumericFilter",
     "UUIDFilter",
+    "TaggableFilter",
     "Page",
     # V2 Core
     "ActionFilter",

zenml/models/v2/base/filter.py

@@ -171,6 +171,8 @@ class BoolFilter(Filter):
 class StrFilter(Filter):
     """Filter for all string fields."""
 
+    json_encode_value: bool = False
+
     ALLOWED_OPS: ClassVar[List[str]] = [
         GenericFilterOps.EQUALS,
         GenericFilterOps.NOT_EQUALS,
@@ -211,16 +213,6 @@ class StrFilter(Filter):
         Raises:
             ValueError: the comparison of the column to a numeric value fails.
         """
-        if self.operation == GenericFilterOps.CONTAINS:
-            return column.like(f"%{self.value}%")
-        if self.operation == GenericFilterOps.STARTSWITH:
-            return column.startswith(f"{self.value}")
-        if self.operation == GenericFilterOps.ENDSWITH:
-            return column.endswith(f"{self.value}")
-        if self.operation == GenericFilterOps.NOT_EQUALS:
-            return column != self.value
-        if self.operation == GenericFilterOps.ONEOF:
-            return column.in_(self.value)
         if self.operation in {
             GenericFilterOps.GT,
             GenericFilterOps.LT,
@@ -254,7 +246,33 @@ class StrFilter(Filter):
                     f"value '{self.value}' (must be numeric): {e}"
                 )
 
-        return column == self.value
+        if self.operation == GenericFilterOps.ONEOF:
+            assert isinstance(self.value, list)
+            # Convert the list of values to a list of json strings
+            json_list = (
+                [json.dumps(v) for v in self.value]
+                if self.json_encode_value
+                else self.value
+            )
+            return column.in_(json_list)
+
+        # Don't convert the value to a json string if the operation is contains
+        # because the quotes around strings will mess with the comparison
+        if self.operation == GenericFilterOps.CONTAINS:
+            return column.like(f"%{self.value}%")
+
+        json_value = (
+            json.dumps(self.value) if self.json_encode_value else self.value
+        )
+
+        if self.operation == GenericFilterOps.STARTSWITH:
+            return column.startswith(f"{json_value}")
+        if self.operation == GenericFilterOps.ENDSWITH:
+            return column.endswith(f"{json_value}")
+        if self.operation == GenericFilterOps.NOT_EQUALS:
+            return column != json_value
+
+        return column == json_value
 
 
 class UUIDFilter(StrFilter):
@@ -733,6 +751,7 @@ class BaseFilter(BaseModel):
         value: Any,
         table: Type[SQLModel],
         column: str,
+        json_encode_value: bool = False,
     ) -> "ColumnElement[bool]":
         """Generate custom filter conditions for a column of a table.
 
@@ -740,6 +759,7 @@ class BaseFilter(BaseModel):
             value: The filter value.
             table: The table which contains the column.
             column: The column name.
+            json_encode_value: Whether to json encode the value.
 
         Returns:
             The query conditions.
@@ -748,6 +768,9 @@ class BaseFilter(BaseModel):
         filter_ = FilterGenerator(table).define_filter(
             column=column, value=value, operator=operator
         )
+        if isinstance(filter_, StrFilter):
+            filter_.json_encode_value = json_encode_value
+
         return filter_.generate_query_conditions(table=table)
 
     @property

zenml/models/v2/base/scoped.py

@@ -464,19 +464,19 @@ class WorkspaceScopedFilter(UserScopedFilter):
         return super().apply_sorting(query=query, table=table)
 
 
-class WorkspaceScopedTaggableFilter(WorkspaceScopedFilter):
-    """Model to enable advanced scoping with workspace and tagging."""
+class TaggableFilter(BaseFilter):
+    """Model to enable filtering and sorting by tags."""
 
     tag: Optional[str] = Field(
         description="Tag to apply to the filter query.", default=None
     )
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *BaseFilter.FILTER_EXCLUDE_FIELDS,
         "tag",
     ]
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *BaseFilter.CUSTOM_SORTING_OPTIONS,
         "tags",
     ]
 

zenml/models/v2/core/artifact.py

@@ -35,7 +35,7 @@ from zenml.models.v2.base.base import (
     BaseResponseMetadata,
     BaseResponseResources,
 )
-from zenml.models.v2.base.scoped import WorkspaceScopedTaggableFilter
+from zenml.models.v2.base.scoped import TaggableFilter
 from zenml.models.v2.core.tag import TagResponse
 
 if TYPE_CHECKING:
@@ -183,14 +183,14 @@ class ArtifactResponse(
 # ------------------ Filter Model ------------------
 
 
-class ArtifactFilter(WorkspaceScopedTaggableFilter):
+class ArtifactFilter(TaggableFilter):
     """Model to enable advanced filtering of artifacts."""
 
     name: Optional[str] = None
     has_custom_name: Optional[bool] = None
 
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         SORT_BY_LATEST_VERSION_KEY,
     ]
 

zenml/models/v2/core/artifact_version.py

@@ -41,12 +41,13 @@ from zenml.logger import get_logger
 from zenml.metadata.metadata_types import MetadataType
 from zenml.models.v2.base.filter import FilterGenerator, StrFilter
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
+    WorkspaceScopedFilter,
     WorkspaceScopedRequest,
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
 )
 from zenml.models.v2.core.artifact import ArtifactResponse
 from zenml.models.v2.core.tag import TagResponse
@@ -469,11 +470,12 @@ class ArtifactVersionResponse(
 # ------------------ Filter Model ------------------
 
 
-class ArtifactVersionFilter(WorkspaceScopedTaggableFilter):
+class ArtifactVersionFilter(WorkspaceScopedFilter, TaggableFilter):
     """Model to enable advanced filtering of artifact versions."""
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.FILTER_EXCLUDE_FIELDS,
+        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "name",
         "only_unused",
         "has_custom_name",
@@ -482,6 +484,15 @@ class ArtifactVersionFilter(WorkspaceScopedTaggableFilter):
         "model_version_id",
         "run_metadata",
     ]
+    CUSTOM_SORTING_OPTIONS = [
+        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
+    ]
+    CLI_EXCLUDE_FIELDS = [
+        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *TaggableFilter.CLI_EXCLUDE_FIELDS,
+    ]
+
     artifact_id: Optional[Union[UUID, str]] = Field(
         default=None,
         description="ID of the artifact to which this version belongs.",
@@ -548,7 +559,7 @@ class ArtifactVersionFilter(WorkspaceScopedTaggableFilter):
         description="Name/ID of a pipeline run that is associated with this "
         "artifact version.",
     )
-    run_metadata: Optional[Dict[str, str]] = Field(
+    run_metadata: Optional[Dict[str, Any]] = Field(
         default=None,
         description="The run_metadata to filter the artifact versions by.",
     )
@@ -672,13 +683,19 @@ class ArtifactVersionFilter(WorkspaceScopedTaggableFilter):
                     RunMetadataResourceSchema.resource_id
                     == ArtifactVersionSchema.id,
                     RunMetadataResourceSchema.resource_type
-                    == MetadataResourceTypes.ARTIFACT_VERSION,
+                    == MetadataResourceTypes.ARTIFACT_VERSION.value,
                     RunMetadataResourceSchema.run_metadata_id
                     == RunMetadataSchema.id,
+                    self.generate_custom_query_conditions_for_column(
+                        value=key,
+                        table=RunMetadataSchema,
+                        column="key",
+                    ),
                     self.generate_custom_query_conditions_for_column(
                         value=value,
                         table=RunMetadataSchema,
                         column="value",
+                        json_encode_value=True,
                     ),
                 )
                 custom_filters.append(additional_filter)

zenml/models/v2/core/model.py

@@ -24,12 +24,13 @@ from zenml.constants import (
     TEXT_FIELD_MAX_LENGTH,
 )
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
+    WorkspaceScopedFilter,
     WorkspaceScopedRequest,
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
 )
 from zenml.utils.pagination_utils import depaginate
 
@@ -322,7 +323,7 @@ class ModelResponse(
 # ------------------ Filter Model ------------------
 
 
-class ModelFilter(WorkspaceScopedTaggableFilter):
+class ModelFilter(WorkspaceScopedFilter, TaggableFilter):
     """Model to enable advanced filtering of all Workspaces."""
 
     name: Optional[str] = Field(
@@ -330,10 +331,19 @@ class ModelFilter(WorkspaceScopedTaggableFilter):
         description="Name of the Model",
     )
 
+    FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
+        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *TaggableFilter.FILTER_EXCLUDE_FIELDS,
+    ]
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.CUSTOM_SORTING_OPTIONS,
+        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         SORT_BY_LATEST_VERSION_KEY,
     ]
+    CLI_EXCLUDE_FIELDS = [
+        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *TaggableFilter.CLI_EXCLUDE_FIELDS,
+    ]
 
     def apply_sorting(
         self,

zenml/models/v2/core/model_version.py

@@ -33,12 +33,13 @@ from zenml.metadata.metadata_types import MetadataType
 from zenml.models.v2.base.filter import AnyQuery
 from zenml.models.v2.base.page import Page
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
+    WorkspaceScopedFilter,
     WorkspaceScopedRequest,
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
 )
 from zenml.models.v2.core.service import ServiceResponse
 from zenml.models.v2.core.tag import TagResponse
@@ -576,13 +577,22 @@ class ModelVersionResponse(
 # ------------------ Filter Model ------------------
 
 
-class ModelVersionFilter(WorkspaceScopedTaggableFilter):
+class ModelVersionFilter(WorkspaceScopedFilter, TaggableFilter):
     """Filter model for model versions."""
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.FILTER_EXCLUDE_FIELDS,
+        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "run_metadata",
     ]
+    CUSTOM_SORTING_OPTIONS = [
+        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
+    ]
+    CLI_EXCLUDE_FIELDS = [
+        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *TaggableFilter.CLI_EXCLUDE_FIELDS,
+    ]
 
     name: Optional[str] = Field(
         default=None,

zenml/models/v2/core/pipeline.py

@@ -34,12 +34,13 @@ from zenml.constants import (
 from zenml.enums import ExecutionStatus
 from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
+    WorkspaceScopedFilter,
     WorkspaceScopedRequest,
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
 )
 from zenml.models.v2.core.tag import TagResponse
 
@@ -256,17 +257,23 @@ class PipelineResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineFilter(WorkspaceScopedTaggableFilter):
+class PipelineFilter(WorkspaceScopedFilter, TaggableFilter):
     """Pipeline filter model."""
 
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.CUSTOM_SORTING_OPTIONS,
+        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         SORT_PIPELINES_BY_LATEST_RUN_KEY,
     ]
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.FILTER_EXCLUDE_FIELDS,
+        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "latest_run_status",
     ]
+    CLI_EXCLUDE_FIELDS = [
+        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *TaggableFilter.CLI_EXCLUDE_FIELDS,
+    ]
 
     name: Optional[str] = Field(
         default=None,

zenml/models/v2/core/pipeline_run.py

@@ -35,12 +35,13 @@ from zenml.constants import STR_FIELD_MAX_LENGTH
 from zenml.enums import ExecutionStatus
 from zenml.metadata.metadata_types import MetadataType
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
+    WorkspaceScopedFilter,
     WorkspaceScopedRequest,
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
 )
 from zenml.models.v2.core.model_version import ModelVersionResponse
 from zenml.models.v2.core.tag import TagResponse
@@ -589,20 +590,21 @@ class PipelineRunResponse(
 # ------------------ Filter Model ------------------
 
 
-class PipelineRunFilter(WorkspaceScopedTaggableFilter):
+class PipelineRunFilter(WorkspaceScopedFilter, TaggableFilter):
     """Model to enable advanced filtering of all Workspaces."""
 
     CUSTOM_SORTING_OPTIONS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.CUSTOM_SORTING_OPTIONS,
+        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
         "tag",
         "stack",
         "pipeline",
         "model",
         "model_version",
     ]
-
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.FILTER_EXCLUDE_FIELDS,
+        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "unlisted",
         "code_repository_id",
         "build_id",
@@ -618,6 +620,11 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
         "templatable",
         "run_metadata",
     ]
+    CLI_EXCLUDE_FIELDS = [
+        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *TaggableFilter.CLI_EXCLUDE_FIELDS,
+    ]
+
     name: Optional[str] = Field(
         default=None,
         description="Name of the Pipeline Run",
@@ -681,7 +688,7 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
         union_mode="left_to_right",
     )
     unlisted: Optional[bool] = None
-    run_metadata: Optional[Dict[str, str]] = Field(
+    run_metadata: Optional[Dict[str, Any]] = Field(
         default=None,
         description="The run_metadata to filter the pipeline runs by.",
     )
@@ -908,13 +915,19 @@ class PipelineRunFilter(WorkspaceScopedTaggableFilter):
                     RunMetadataResourceSchema.resource_id
                     == PipelineRunSchema.id,
                     RunMetadataResourceSchema.resource_type
-                    == MetadataResourceTypes.PIPELINE_RUN,
+                    == MetadataResourceTypes.PIPELINE_RUN.value,
                     RunMetadataResourceSchema.run_metadata_id
                     == RunMetadataSchema.id,
+                    self.generate_custom_query_conditions_for_column(
+                        value=key,
+                        table=RunMetadataSchema,
+                        column="key",
+                    ),
                     self.generate_custom_query_conditions_for_column(
                         value=value,
                         table=RunMetadataSchema,
                         column="value",
+                        json_encode_value=True,
                     ),
                 )
                 custom_filters.append(additional_filter)

zenml/models/v2/core/run_template.py

@@ -33,12 +33,13 @@ from zenml.constants import STR_FIELD_MAX_LENGTH, TEXT_FIELD_MAX_LENGTH
 from zenml.enums import ExecutionStatus
 from zenml.models.v2.base.base import BaseUpdate
 from zenml.models.v2.base.scoped import (
+    TaggableFilter,
+    WorkspaceScopedFilter,
     WorkspaceScopedRequest,
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
     WorkspaceScopedResponseResources,
-    WorkspaceScopedTaggableFilter,
 )
 from zenml.models.v2.core.code_reference import (
     CodeReferenceResponse,
@@ -307,11 +308,12 @@ class RunTemplateResponse(
 # ------------------ Filter Model ------------------
 
 
-class RunTemplateFilter(WorkspaceScopedTaggableFilter):
+class RunTemplateFilter(WorkspaceScopedFilter, TaggableFilter):
     """Model for filtering of run templates."""
 
     FILTER_EXCLUDE_FIELDS: ClassVar[List[str]] = [
-        *WorkspaceScopedTaggableFilter.FILTER_EXCLUDE_FIELDS,
+        *WorkspaceScopedFilter.FILTER_EXCLUDE_FIELDS,
+        *TaggableFilter.FILTER_EXCLUDE_FIELDS,
         "code_repository_id",
         "stack_id",
         "build_id",
@@ -320,6 +322,14 @@ class RunTemplateFilter(WorkspaceScopedTaggableFilter):
         "pipeline",
         "stack",
     ]
+    CUSTOM_SORTING_OPTIONS = [
+        *WorkspaceScopedFilter.CUSTOM_SORTING_OPTIONS,
+        *TaggableFilter.CUSTOM_SORTING_OPTIONS,
+    ]
+    CLI_EXCLUDE_FIELDS = [
+        *WorkspaceScopedFilter.CLI_EXCLUDE_FIELDS,
+        *TaggableFilter.CLI_EXCLUDE_FIELDS,
+    ]
 
     name: Optional[str] = Field(
         default=None,

zenml/models/v2/core/step_run.py

@@ -16,6 +16,7 @@
 from datetime import datetime
 from typing import (
     TYPE_CHECKING,
+    Any,
     ClassVar,
     Dict,
     List,
@@ -574,7 +575,7 @@ class StepRunFilter(WorkspaceScopedFilter):
         default=None,
         description="Name/ID of the model associated with the step run.",
     )
-    run_metadata: Optional[Dict[str, str]] = Field(
+    run_metadata: Optional[Dict[str, Any]] = Field(
         default=None,
         description="The run_metadata to filter the step runs by.",
     )
@@ -619,13 +620,19 @@ class StepRunFilter(WorkspaceScopedFilter):
                 additional_filter = and_(
                     RunMetadataResourceSchema.resource_id == StepRunSchema.id,
                     RunMetadataResourceSchema.resource_type
-                    == MetadataResourceTypes.STEP_RUN,
+                    == MetadataResourceTypes.STEP_RUN.value,
                     RunMetadataResourceSchema.run_metadata_id
                     == RunMetadataSchema.id,
+                    self.generate_custom_query_conditions_for_column(
+                        value=key,
+                        table=RunMetadataSchema,
+                        column="key",
+                    ),
                     self.generate_custom_query_conditions_for_column(
                         value=value,
                         table=RunMetadataSchema,
                         column="value",
+                        json_encode_value=True,
                     ),
                 )
                 custom_filters.append(additional_filter)

zenml/pipelines/pipeline_definition.py

@@ -407,17 +407,31 @@ class Pipeline:
         return self
 
     @property
-    def requires_parameters(self) -> bool:
-        """If the pipeline entrypoint requires parameters.
+    def required_parameters(self) -> List[str]:
+        """List of required parameters for the pipeline entrypoint.
 
         Returns:
-            If the pipeline entrypoint requires parameters.
+            List of required parameters for the pipeline entrypoint.
         """
         signature = inspect.signature(self.entrypoint, follow_wrapped=True)
-        return any(
-            parameter.default is inspect.Parameter.empty
+        return [
+            parameter.name
             for parameter in signature.parameters.values()
-        )
+            if parameter.default is inspect.Parameter.empty
+        ]
+
+    @property
+    def missing_parameters(self) -> List[str]:
+        """List of missing parameters for the pipeline entrypoint.
+
+        Returns:
+            List of missing parameters for the pipeline entrypoint.
+        """
+        available_parameters = set(self.configuration.parameters or {})
+        if params_from_file := self._from_config_file.get("parameters", None):
+            available_parameters.update(params_from_file)
+
+        return list(set(self.required_parameters) - available_parameters)
 
     @property
     def is_prepared(self) -> bool:
@@ -1412,7 +1426,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
         except ValidationError as e:
             raise ValueError(
                 "Invalid or missing pipeline function entrypoint arguments. "
-                "Only JSON serializable inputs are allowed as pipeline inputs."
+                "Only JSON serializable inputs are allowed as pipeline inputs. "
                 "Check out the pydantic error above for more details."
             ) from e
 
@@ -1427,15 +1441,17 @@ To avoid this consider setting pipeline parameters only in one place (config or
                 requires parameters.
         """
         if not self.is_prepared:
-            if self.requires_parameters:
+            if missing_parameters := self.missing_parameters:
                 raise RuntimeError(
                     f"Failed while trying to prepare pipeline {self.name}. "
                     "The entrypoint function of the pipeline requires "
-                    "arguments. Please prepare the pipeline by calling "
-                    "`pipeline_instance.prepare(...)` and try again."
+                    "arguments which have not been configured yet: "
+                    f"{missing_parameters}. Please provide those parameters by "
+                    "calling `pipeline_instance.configure(parameters=...)` or "
+                    "by calling `pipeline_instance.prepare(...)` and try again."
                 )
-            else:
-                self.prepare()
+
+            self.prepare()
 
     def create_run_template(
         self, name: str, **kwargs: Any

zenml/zen_stores/schemas/artifact_schemas.py

@@ -18,7 +18,8 @@ from uuid import UUID
 
 from pydantic import ValidationError
 from sqlalchemy import TEXT, Column, UniqueConstraint
-from sqlmodel import Field, Relationship
+from sqlalchemy.orm import object_session
+from sqlmodel import Field, Relationship, desc, select
 
 from zenml.config.source import Source
 from zenml.enums import (
@@ -90,6 +91,32 @@ class ArtifactSchema(NamedSchema, table=True):
         ),
     )
 
+    @property
+    def latest_version(self) -> Optional["ArtifactVersionSchema"]:
+        """Fetch the latest version for this artifact.
+
+        Raises:
+            RuntimeError: If no session for the schema exists.
+
+        Returns:
+            The latest version for this artifact.
+        """
+        if session := object_session(self):
+            return (
+                session.execute(
+                    select(ArtifactVersionSchema)
+                    .where(ArtifactVersionSchema.artifact_id == self.id)
+                    .order_by(desc(ArtifactVersionSchema.created))
+                    .limit(1)
+                )
+                .scalars()
+                .one_or_none()
+            )
+        else:
+            raise RuntimeError(
+                "Missing DB session to fetch latest version for artifact."
+            )
+
     @classmethod
     def from_request(
         cls,
@@ -127,9 +154,9 @@ class ArtifactSchema(NamedSchema, table=True):
             The created `ArtifactResponse`.
         """
         latest_id, latest_name = None, None
-        if self.versions:
-            latest_version = max(self.versions, key=lambda x: x.created)
-            latest_id, latest_name = latest_version.id, latest_version.version
+        if latest_version := self.latest_version:
+            latest_id = latest_version.id
+            latest_name = latest_version.version
 
         # Create the body of the model
         body = ArtifactResponseBody(

zenml/zen_stores/schemas/model_schemas.py

@@ -24,7 +24,8 @@ from sqlalchemy import (
     Column,
     UniqueConstraint,
 )
-from sqlmodel import Field, Relationship
+from sqlalchemy.orm import object_session
+from sqlmodel import Field, Relationship, desc, select
 
 from zenml.enums import (
     ArtifactType,
@@ -126,6 +127,32 @@ class ModelSchema(NamedSchema, table=True):
         sa_relationship_kwargs={"cascade": "delete"},
     )
 
+    @property
+    def latest_version(self) -> Optional["ModelVersionSchema"]:
+        """Fetch the latest version for this model.
+
+        Raises:
+            RuntimeError: If no session for the schema exists.
+
+        Returns:
+            The latest version for this model.
+        """
+        if session := object_session(self):
+            return (
+                session.execute(
+                    select(ModelVersionSchema)
+                    .where(ModelVersionSchema.model_id == self.id)
+                    .order_by(desc(ModelVersionSchema.number))
+                    .limit(1)
+                )
+                .scalars()
+                .one_or_none()
+            )
+        else:
+            raise RuntimeError(
+                "Missing DB session to fetch latest version for model."
+            )
+
     @classmethod
     def from_request(cls, model_request: ModelRequest) -> "ModelSchema":
         """Convert an `ModelRequest` to an `ModelSchema`.
@@ -169,11 +196,9 @@ class ModelSchema(NamedSchema, table=True):
         """
         tags = [tag.to_model() for tag in self.tags]
 
-        if self.model_versions:
-            version_numbers = [mv.number for mv in self.model_versions]
-            latest_version_idx = version_numbers.index(max(version_numbers))
-            latest_version_name = self.model_versions[latest_version_idx].name
-            latest_version_id = self.model_versions[latest_version_idx].id
+        if latest_version := self.latest_version:
+            latest_version_name = latest_version.name
+            latest_version_id = latest_version.id
         else:
             latest_version_name = None
             latest_version_id = None

zenml/zen_stores/schemas/pipeline_run_schemas.py

@@ -208,7 +208,7 @@ class PipelineRunSchema(NamedSchema, RunMetadataInterface, table=True):
     stack: Optional["StackSchema"] = Relationship()
     build: Optional["PipelineBuildSchema"] = Relationship()
     schedule: Optional["ScheduleSchema"] = Relationship()
-    pipeline: Optional["PipelineSchema"] = Relationship(back_populates="runs")
+    pipeline: Optional["PipelineSchema"] = Relationship()
     trigger_execution: Optional["TriggerExecutionSchema"] = Relationship()
 
     services: List["ServiceSchema"] = Relationship(

zenml/zen_stores/schemas/pipeline_schemas.py

@@ -17,7 +17,8 @@ from typing import TYPE_CHECKING, Any, List, Optional
 from uuid import UUID
 
 from sqlalchemy import TEXT, Column, UniqueConstraint
-from sqlmodel import Field, Relationship
+from sqlalchemy.orm import object_session
+from sqlmodel import Field, Relationship, desc, select
 
 from zenml.enums import TaggableResourceTypes
 from zenml.models import (
@@ -85,10 +86,6 @@ class PipelineSchema(NamedSchema, table=True):
     schedules: List["ScheduleSchema"] = Relationship(
         back_populates="pipeline",
     )
-    runs: List["PipelineRunSchema"] = Relationship(
-        back_populates="pipeline",
-        sa_relationship_kwargs={"order_by": "PipelineRunSchema.created"},
-    )
     builds: List["PipelineBuildSchema"] = Relationship(
         back_populates="pipeline"
     )
@@ -105,6 +102,34 @@ class PipelineSchema(NamedSchema, table=True):
         ),
     )
 
+    @property
+    def latest_run(self) -> Optional["PipelineRunSchema"]:
+        """Fetch the latest run for this pipeline.
+
+        Raises:
+            RuntimeError: If no session for the schema exists.
+
+        Returns:
+            The latest run for this pipeline.
+        """
+        from zenml.zen_stores.schemas import PipelineRunSchema
+
+        if session := object_session(self):
+            return (
+                session.execute(
+                    select(PipelineRunSchema)
+                    .where(PipelineRunSchema.pipeline_id == self.id)
+                    .order_by(desc(PipelineRunSchema.created))
+                    .limit(1)
+                )
+                .scalars()
+                .one_or_none()
+            )
+        else:
+            raise RuntimeError(
+                "Missing DB session to fetch latest run for pipeline."
+            )
+
     @classmethod
     def from_request(
         cls,
@@ -141,10 +166,12 @@ class PipelineSchema(NamedSchema, table=True):
         Returns:
             The created PipelineResponse.
         """
+        latest_run = self.latest_run
+
         body = PipelineResponseBody(
             user=self.user.to_model() if self.user else None,
-            latest_run_id=self.runs[-1].id if self.runs else None,
-            latest_run_status=self.runs[-1].status if self.runs else None,
+            latest_run_id=latest_run.id if latest_run else None,
+            latest_run_status=latest_run.status if latest_run else None,
             created=self.created,
             updated=self.updated,
         )
@@ -158,7 +185,7 @@ class PipelineSchema(NamedSchema, table=True):
 
         resources = None
         if include_resources:
-            latest_run_user = self.runs[-1].user if self.runs else None
+            latest_run_user = latest_run.user if latest_run else None
 
             resources = PipelineResponseResources(
                 latest_run_user=latest_run_user.to_model()

zenml/zen_stores/schemas/run_template_schemas.py

@@ -18,7 +18,8 @@ from uuid import UUID
 
 from sqlalchemy import Column, String, UniqueConstraint
 from sqlalchemy.dialects.mysql import MEDIUMTEXT
-from sqlmodel import Field, Relationship
+from sqlalchemy.orm import object_session
+from sqlmodel import Field, Relationship, col, desc, select
 
 from zenml.constants import MEDIUMTEXT_MAX_LENGTH
 from zenml.enums import TaggableResourceTypes
@@ -99,17 +100,6 @@ class RunTemplateSchema(BaseSchema, table=True):
         }
     )
 
-    runs: List["PipelineRunSchema"] = Relationship(
-        sa_relationship_kwargs={
-            "primaryjoin": "RunTemplateSchema.id==PipelineDeploymentSchema.template_id",
-            "secondaryjoin": "PipelineDeploymentSchema.id==PipelineRunSchema.deployment_id",
-            "secondary": "pipeline_deployment",
-            "cascade": "delete",
-            "viewonly": True,
-            "order_by": "PipelineRunSchema.created",
-        }
-    )
-
     tags: List["TagSchema"] = Relationship(
         sa_relationship_kwargs=dict(
             primaryjoin=f"and_(foreign(TagResourceSchema.resource_type)=='{TaggableResourceTypes.RUN_TEMPLATE.value}', foreign(TagResourceSchema.resource_id)==RunTemplateSchema.id)",
@@ -120,6 +110,42 @@ class RunTemplateSchema(BaseSchema, table=True):
         ),
     )
 
+    @property
+    def latest_run(self) -> Optional["PipelineRunSchema"]:
+        """Fetch the latest run for this template.
+
+        Raises:
+            RuntimeError: If no session for the schema exists.
+
+        Returns:
+            The latest run for this template.
+        """
+        from zenml.zen_stores.schemas import (
+            PipelineDeploymentSchema,
+            PipelineRunSchema,
+        )
+
+        if session := object_session(self):
+            return (
+                session.execute(
+                    select(PipelineRunSchema)
+                    .join(
+                        PipelineDeploymentSchema,
+                        col(PipelineDeploymentSchema.id)
+                        == col(PipelineRunSchema.deployment_id),
+                    )
+                    .where(PipelineDeploymentSchema.template_id == self.id)
+                    .order_by(desc(PipelineRunSchema.created))
+                    .limit(1)
+                )
+                .scalars()
+                .one_or_none()
+            )
+        else:
+            raise RuntimeError(
+                "Missing DB session to fetch latest run for template."
+            )
+
     @classmethod
     def from_request(
         cls,
@@ -184,13 +210,15 @@ class RunTemplateSchema(BaseSchema, table=True):
         ):
             runnable = True
 
+        latest_run = self.latest_run
+
         body = RunTemplateResponseBody(
             user=self.user.to_model() if self.user else None,
             created=self.created,
             updated=self.updated,
             runnable=runnable,
-            latest_run_id=self.runs[-1].id if self.runs else None,
-            latest_run_status=self.runs[-1].status if self.runs else None,
+            latest_run_id=latest_run.id if latest_run else None,
+            latest_run_status=latest_run.status if latest_run else None,
         )
 
         metadata = None

{zenml_nightly-0.73.0.dev20250205.dist-info → zenml_nightly-0.73.0.dev20250206.dist-info}/METADATA

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

{zenml_nightly-0.73.0.dev20250205.dist-info → zenml_nightly-0.73.0.dev20250206.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=BRNvbbNFEwdyrIEopU926a0ad2FzMvnprZW8ctT0PFw,19
+zenml/VERSION,sha256=ji5Aj9EhfhLiyX0DtPoxdFiamKZSRAvGE4VKg4u9TaA,19
 zenml/__init__.py,sha256=SkMObQA41ajqdZqGErN00S1Vf3KAxpLvbZ-OBy5uYoo,2130
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -52,10 +52,10 @@ zenml/cli/stack_components.py,sha256=7RXVCJTX6sU-CiyDeu4kGYPJZ7QRQjDTptsvl__sI_A
 zenml/cli/tag.py,sha256=Co-AL1eNZMf_AsqlcnJpoZ_k7UKa0l5X_g_TLpJdGes,4884
 zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=fTuRworQahst_j78qPYTtgciUeUOxwo7efiyPwmj2tI,13075
-zenml/cli/utils.py,sha256=kqTFuEyXgx0eP_7mtWwmDwF3JvDSQ7dnOMhlNYSSJfM,86594
+zenml/cli/utils.py,sha256=oQm8clYsuATzEjXJNVMwmdTiygOwxErvUGONPIxb7no,86478
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
 zenml/cli/workspace.py,sha256=bp02aXou574ToWPD8OAIB_cg3mvpE011H8aMKegT-nU,2970
-zenml/client.py,sha256=B2xEX39nePNAl5TcFVyvIMYBQqAba5IQ5BFhLC2enxQ,283388
+zenml/client.py,sha256=VGpvUNfs6D_7JthAjocEO6pDNfjeNkWwNudWVYrbgqo,283571
 zenml/client_lazy_loader.py,sha256=MOBgS1ITYqGvPUnWQ6edn9s8Hr_72YfWbwEIfHKUr9g,7104
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=Id6VjbUu8N3ZpNvBGhOgbahtoMiCAtYXed3G7YQ_iAc,5225
@@ -615,20 +615,20 @@ zenml/model_deployers/__init__.py,sha256=oVBLtTfrNenl5OI1iqtQUvJ0vpocRVUN_HIt8qp
 zenml/model_deployers/base_model_deployer.py,sha256=Xg5lxBFYM41vqxQhaB54Dxu_zLCyPDgqwrTyMcAxiS4,24609
 zenml/model_registries/__init__.py,sha256=wA9Vzo0w_e9zuXOVURB9w8oMLSnTaimXcxg_Nb7O3b0,1238
 zenml/model_registries/base_model_registry.py,sha256=nK__C6IfoIRxnOHA9qyPErLvIsbakyac4ZltUZ8t3tk,17526
-zenml/models/__init__.py,sha256=zKOStbjGZPfDJcsIYZdYZitTFQWISUciN8poXCNbeEM,21614
+zenml/models/__init__.py,sha256=sLMSaAWnoHYKAE0Tq8vi5o2k0fVeYiM7z72TwQyUZbI,21640
 zenml/models/v2/__init__.py,sha256=LGMIUJi19sOsvo54roZSQwDp_czNQYtenqd_frTLIhU,613
 zenml/models/v2/base/__init__.py,sha256=knhroJ2h0uHBCGzAiBBGJEiuhEA3cwI6XYBRIyXdbkQ,613
 zenml/models/v2/base/base.py,sha256=tXzNTIwGNz7MEgfii3-6ASBNFg4SkR9D9ZM48tlNr84,16242
 zenml/models/v2/base/base_plugin_flavor.py,sha256=BfPL4gm5i7ad7_vuRmPtC-rE2kl0W_8SmpWme8Akv1c,2608
-zenml/models/v2/base/filter.py,sha256=uCZa3MFdEmUL91DYa43vZVuHYBQDC-z3gZWUzzqd63U,36990
+zenml/models/v2/base/filter.py,sha256=ScuQusfFOPUU52eAiLDditU7_IFcD6CeFVqrJwP6H20,37783
 zenml/models/v2/base/page.py,sha256=yBpWB-8iUmyRwXKSJvX3v9SB7WOnyRa25O8SAss24xk,2601
-zenml/models/v2/base/scoped.py,sha256=fOZpuTTUjkg2xE8LvPFUP8lpU9HC-chSQ8xNpmWAW-c,17578
+zenml/models/v2/base/scoped.py,sha256=FrM-pSE2gKoVffMg-Zoei6lu4C4Rvpp-icIdbk6q9dE,17516
 zenml/models/v2/core/__init__.py,sha256=LGMIUJi19sOsvo54roZSQwDp_czNQYtenqd_frTLIhU,613
 zenml/models/v2/core/action.py,sha256=6n6Lep_2-OVUDCWs6eKiUHFKLvcoMjHWP4ZSbua1JRY,7911
 zenml/models/v2/core/action_flavor.py,sha256=DlGlJTOcgLSDYAsrbAyXOJVdo7X2GYTFQTTEG5BEf-Q,1718
 zenml/models/v2/core/api_key.py,sha256=Zv2N4NDraZUIrBOW1sSQtVp0Ik6zpdTFymYCOo17c-4,11534
-zenml/models/v2/core/artifact.py,sha256=VZDjn3xjwafh3M6Jq38wDKkzdYI3t3eOsIFnM6STMf8,7655
-zenml/models/v2/core/artifact_version.py,sha256=CWVLWzmeyvQrHRevTPkFBAcJTxIlwMx3drcV3f2bJ7k,23134
+zenml/models/v2/core/artifact.py,sha256=dmw36qIwtPDptU09nWGlwXhE0v0xXQ_4naIvndKh1ec,7610
+zenml/models/v2/core/artifact_version.py,sha256=0eJFj9DJ7T6yfNp6W0qqjLwKIzfNOeiENAk9tk51I7o,23731
 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=W4BuE8oterWPmtbCUCtOVsw8RfaOgiZ5aefuQHxL8dk,5580
@@ -638,16 +638,16 @@ zenml/models/v2/core/event_source.py,sha256=kOexFvxj3iqsQGiGHPe549v9cSTo2GfsPCK0
 zenml/models/v2/core/event_source_flavor.py,sha256=TgrGkypnJUutMvvHE7Y9MbUiEeNXZxitc3eYlIbbsas,2077
 zenml/models/v2/core/flavor.py,sha256=tcn2GDbkOyYjXTor_7_kR7FuLwcXjlfzvAV2rU1NVM4,12806
 zenml/models/v2/core/logs.py,sha256=UhaPOTJD7IFKH8YozsaVgXYNNDHre8amheIm5JMZYQU,5824
-zenml/models/v2/core/model.py,sha256=My56dONvNkN95C121M0RS3lB_oQIRJpd9z_mHHZkU6A,11572
-zenml/models/v2/core/model_version.py,sha256=nDj8A4BUMBFwkKXF8VW1QiDwP6MXqrP6P9pIKB5XjXQ,20735
+zenml/models/v2/core/model.py,sha256=yok73T2D4hzYGP0nhDs5mq-1LKA9iol_k5DQyCVwf1o,11918
+zenml/models/v2/core/model_version.py,sha256=q3fTYRSLR7i_84VFWqCmcm4FqGhwtlq2Wfhtm6G0u8M,21062
 zenml/models/v2/core/model_version_artifact.py,sha256=ZyupuEsfPSjgaHDIUOojwJepJt1LTXDoUyE2h9htB_s,9086
 zenml/models/v2/core/model_version_pipeline_run.py,sha256=JbPZZEQvOK9I32htkWdAONy7gvtoz_Jo45B5gYrh2vs,7036
-zenml/models/v2/core/pipeline.py,sha256=TGsxPvFzGwHebEVNyNc2kHCQQlMB8CCWWveKXDkACYY,11817
+zenml/models/v2/core/pipeline.py,sha256=cql05iz2jqrlyRfn3jB4iQ_AgpHapD6Sr5Kd_6Q1W30,12044
 zenml/models/v2/core/pipeline_build.py,sha256=bjSvmNnTF9xAycYfmTHO2wzgR-ZVWyy5V5-9ajZUxag,16399
 zenml/models/v2/core/pipeline_deployment.py,sha256=6PbpnEUAbTT-_jPmyNtKuTJHFGdGLD2h5Zu_pXIHW1I,12108
-zenml/models/v2/core/pipeline_run.py,sha256=HwZtUGDdjfxUmlbTdkI6Fnvlt0jiLk2aatwYwbhjonI,32699
+zenml/models/v2/core/pipeline_run.py,sha256=mmr3EP5jWq7sQ5FnUTD4kXYWY-yF5x6SW9Ei2a78Stc,33195
 zenml/models/v2/core/run_metadata.py,sha256=YDiPVRzTUMVHqW4T_Ke4GJMdscp0aS_g8I4P8n70szc,2436
-zenml/models/v2/core/run_template.py,sha256=A_dezNA9o1uJW-o1B9fYaSSWHwwglJYP3jvhWhGcJK4,12971
+zenml/models/v2/core/run_template.py,sha256=R2ADxFbBi_zmJE7FA4M0DStIc-mfTiOMuVosniZH9XA,13298
 zenml/models/v2/core/schedule.py,sha256=JNYTGPXDPMz7A9OjhnHIXyOLQuoFnrzOwChGibz2xQY,10774
 zenml/models/v2/core/secret.py,sha256=LoY5BMZIrU6KYa39layHt6cbAnmMR7yXizv_1Qqse6Q,11460
 zenml/models/v2/core/server_settings.py,sha256=al6LAXhoQHsBiTv2qlwFn1AlI2F8miogGB4xjDE6KeA,6267
@@ -655,7 +655,7 @@ zenml/models/v2/core/service.py,sha256=lwCDOFk-OXmfhB_bOOJy7rhbkR40tCScQsWnllgL7
 zenml/models/v2/core/service_account.py,sha256=-1c9Et9Ma4_OHOZxIHjVnLJAaLLxhpoLEyGKkwc1Yx8,6619
 zenml/models/v2/core/service_connector.py,sha256=-Fgi2bilbG99Fo0TWdelpTHMqi8l7E5M5S757b0VjdQ,37411
 zenml/models/v2/core/stack.py,sha256=K5WVJ3vqjoHUlZ4Ezx-Bff3aSgTiBwpH7eViGVSx4YI,12083
-zenml/models/v2/core/step_run.py,sha256=6plyAHt5r9z5TQHh_g_USx8zkO41US8GeeO745iDf-U,19216
+zenml/models/v2/core/step_run.py,sha256=3LyE6z3K4cQgNC2B7nkyWHSUlaRNdLv-abAcxpnd1MM,19494
 zenml/models/v2/core/tag.py,sha256=gQNkxmcksMxx5RJf1U_ug5-VxPhfaS11j--J3dl6muk,3534
 zenml/models/v2/core/tag_resource.py,sha256=H7CPj9oaUpym8vSkwuYSJ6_rWfHMEKVRedPw2kyxmII,2640
 zenml/models/v2/core/trigger.py,sha256=CDbKrJnHoYgYzFlS6gYIjrWihMSQkRyUwdUHNSgHSxY,12808
@@ -695,7 +695,7 @@ zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,8
 zenml/pipelines/build_utils.py,sha256=DkID1YnRYkw569uU-gidpF8WR8E7K55_wn3CzRPQ3Cs,27562
 zenml/pipelines/pipeline_context.py,sha256=V_p-F9W7cBIlTjS0iv5-uJYMzaOj8bAUkc_uNhQgBms,3579
 zenml/pipelines/pipeline_decorator.py,sha256=FIbflYOMavbuyGmqsx3F5zZgg0oXMTi1eAcGXciljOs,4293
-zenml/pipelines/pipeline_definition.py,sha256=M4r2EqpXzAAPG_asK76fYE855U4nlCkZe1To-D4yDM4,56206
+zenml/pipelines/pipeline_definition.py,sha256=Pi4OGs7-WXbsm8_zczY5lpQEyLJPqkKZHjXAKfnk3HY,56936
 zenml/pipelines/run_utils.py,sha256=4KuHIQFtLXTZNQBScTEkIG5pqNtu6xGm6UZT7ptyyKs,11623
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
@@ -1234,7 +1234,7 @@ zenml/zen_stores/rest_zen_store.py,sha256=tB890hOnYaixDKWr7xddpNsNI0GxLV7D4PQd-T
 zenml/zen_stores/schemas/__init__.py,sha256=vMb0Pyf94aMWZDWpPKuRaxNoMsTp9DZNKSQScm9Vd5I,4339
 zenml/zen_stores/schemas/action_schemas.py,sha256=uAWlV3Uy8LERBlu7PHCsWEEF4ww6olsVetzVBWmdDSc,6298
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=d_ewTrIJpRjLWupOZFpWcPSlRiHxiBsZJc_qSU7tMMk,7262
-zenml/zen_stores/schemas/artifact_schemas.py,sha256=_hXytHqa0BYrBLgZt8RdAid1n291d8wQ0HxNMIAjlJQ,14317
+zenml/zen_stores/schemas/artifact_schemas.py,sha256=7xzoiDpMvGu02YV4JGQ7If-5XKpW3-SErs3yOQTK62w,15176
 zenml/zen_stores/schemas/artifact_visualization_schemas.py,sha256=_gMNjOF0oJ0ruW5Ua86hDocvO4MT3ACAIx_BwFBJFWk,3693
 zenml/zen_stores/schemas/base_schemas.py,sha256=-MDYnS4ZLDe6x75nTXuVrOSbcYOuECLNk-yxjfEdJh4,2043
 zenml/zen_stores/schemas/code_repository_schemas.py,sha256=K2EFvu_bkF9-cDAVYpMntNidSu-zb3By7CVq6rLj97c,7489
@@ -1244,13 +1244,13 @@ zenml/zen_stores/schemas/device_schemas.py,sha256=voVBknm14NmFpij1EGb_HhVlUng0cg
 zenml/zen_stores/schemas/event_source_schemas.py,sha256=qhkyKg3zl7pFaxDYcXsDvJEIQAGWzWDrH1S88-cbxFQ,6391
 zenml/zen_stores/schemas/flavor_schemas.py,sha256=cEKHrQp4fr16SDjOqBubM-jAbq4dK3MVyZSe-70e-ME,5054
 zenml/zen_stores/schemas/logs_schemas.py,sha256=qv6fs3JiVgzlmTXJqb_gG5NsU5q_50e0167JiWIxR14,3588
-zenml/zen_stores/schemas/model_schemas.py,sha256=--ep8SeYEjVb3k8hn_lqLVryz9Errh6AIZiW88DEaXU,24770
+zenml/zen_stores/schemas/model_schemas.py,sha256=95q5OaQ9DFmTno76wLaLmeBj3cePrVju20flWiJHPOk,25460
 zenml/zen_stores/schemas/pipeline_build_schemas.py,sha256=QK3sVPYhumUx_x-H3YZ-RfnsxGhKfkMGdP5FC9WrQgU,5718
 zenml/zen_stores/schemas/pipeline_deployment_schemas.py,sha256=fdsROPBE1QrWKvSkaZYz2Sc1MwXogAembs8TfZML3ks,10201
-zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=vGH8-onUSmXwZz-qabZjl2F5bJ06pwVSU5vVr5WtdT8,18067
-zenml/zen_stores/schemas/pipeline_schemas.py,sha256=G8YH5g_kp4QHcijchO5SUNvePXErqoF9qNNzgDg8K_k,6366
+zenml/zen_stores/schemas/pipeline_run_schemas.py,sha256=ylodIinwKGqKCIcZaKJwiA27g0rkNjKY3DLx_E2lylo,18046
+zenml/zen_stores/schemas/pipeline_schemas.py,sha256=tUgBics0QOszIN870_-ADcBHINH8v32-CLjPW0vHzMg,7159
 zenml/zen_stores/schemas/run_metadata_schemas.py,sha256=yDDWEl9q4CExCL47uf72fDEodKNqA60PptBPydCJTwk,3619
-zenml/zen_stores/schemas/run_template_schemas.py,sha256=W3ADwqG6xe9a-zhtvtTREn48onE5uYi6YMI26yU37Xo,8950
+zenml/zen_stores/schemas/run_template_schemas.py,sha256=vnwAHPkjpHyRSSZmKjBemICUOPHSJWuKJmn9JqOstmM,9760
 zenml/zen_stores/schemas/schedule_schema.py,sha256=zLh2LlgRWl-m1ANLpUnwh8LEpKykIR8JpJS7angMGW4,8169
 zenml/zen_stores/schemas/schema_utils.py,sha256=Xahifq2fJ5szXCM00ZZ6461Div9Suatzl6sy9hVhPkk,3612
 zenml/zen_stores/schemas/secret_schemas.py,sha256=H-DGSmVmsH9OPh4BOBC0QXMjV6u-KiKT_KDbfOtkt8M,9889
@@ -1276,8 +1276,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=tKOghX2Wa0f0xSYu_ReqHgCwslvHrBhE3RePJG2tcA0,415921
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.73.0.dev20250205.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.73.0.dev20250205.dist-info/METADATA,sha256=4aKyp4ElvyYZyzp3o1VprHUTmBBphTITDQ4BFQVD_NM,21428
-zenml_nightly-0.73.0.dev20250205.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-zenml_nightly-0.73.0.dev20250205.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.73.0.dev20250205.dist-info/RECORD,,
+zenml_nightly-0.73.0.dev20250206.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.73.0.dev20250206.dist-info/METADATA,sha256=ENIa7gJ176ZakUYpcBlt4HELhVscA_YSwNd917ZeNdg,21428
+zenml_nightly-0.73.0.dev20250206.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+zenml_nightly-0.73.0.dev20250206.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.73.0.dev20250206.dist-info/RECORD,,