zenml-nightly 0.73.0.dev20250205__py3-none-any.whl → 0.74.0.dev20250207__py3-none-any.whl

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_server/dashboard/assets/{404-c8OuXDAT.js → 404-0leSZ7Vv.js}

@@ -1 +1 @@
-import{j as e}from"./@radix-DeK6qiuw.js";import{f as s,r as t}from"./index-CxO6541P.js";import{E as r}from"./EmptyState-BzdlCwp3.js";import{S as a}from"./help-Cc9bBIJH.js";import{L as o}from"./@react-router-B3Z5rLr2.js";import"./@tanstack-DT5WLu9C.js";import"./@reactflow-6JPoencd.js";function d(){return e.jsx("div",{className:"flex min-h-screen w-full flex-col",children:e.jsx(r,{icon:e.jsx(a,{className:"h-[120px] w-[120px] fill-neutral-300"}),children:e.jsxs("div",{className:"text-center",children:[e.jsx("h1",{className:"mb-2 text-display-xs font-semibold",children:"We can't find the page you are looking for"}),e.jsx("p",{className:"text-lg text-theme-text-secondary",children:"You can try typing a different URL or we can bring you back to your Homepage."}),e.jsx("div",{className:"mt-5 flex justify-center",children:e.jsx(s,{size:"md",asChild:!0,children:e.jsx(o,{className:"w-min self-center whitespace-nowrap",to:t.home,children:e.jsx("span",{className:"px-0.5",children:"Go to Home"})})})})]})})})}export{d as default};
+import{j as e}from"./@radix-B7OY3d6N.js";import{f as s,r as t}from"./index-D25WcEJJ.js";import{E as r}from"./EmptyState-C8-mkRwm.js";import{S as a}from"./help-CgM5zwtY.js";import{L as o}from"./@react-router-CiF9laDg.js";import"./@tanstack-W1pLDzVg.js";import"./@reactflow-DgwdIYyq.js";function d(){return e.jsx("div",{className:"flex min-h-screen w-full flex-col",children:e.jsx(r,{icon:e.jsx(a,{className:"h-[120px] w-[120px] fill-neutral-300"}),children:e.jsxs("div",{className:"text-center",children:[e.jsx("h1",{className:"mb-2 text-display-xs font-semibold",children:"We can't find the page you are looking for"}),e.jsx("p",{className:"text-lg text-theme-text-secondary",children:"You can try typing a different URL or we can bring you back to your Homepage."}),e.jsx("div",{className:"mt-5 flex justify-center",children:e.jsx(s,{size:"md",asChild:!0,children:e.jsx(o,{className:"w-min self-center whitespace-nowrap",to:t.home,children:e.jsx("span",{className:"px-0.5",children:"Go to Home"})})})})]})})})}export{d as default};