zenml-nightly 0.80.1.dev20250407__py3-none-any.whl → 0.80.1.dev20250409__py3-none-any.whl

zenml/steps/base_step.py

@@ -605,9 +605,9 @@ class BaseStep:
         on_failure: Optional["HookSpecification"] = None,
         on_success: Optional["HookSpecification"] = None,
         model: Optional["Model"] = None,
-        merge: bool = True,
         retry: Optional[StepRetryConfig] = None,
         substitutions: Optional[Dict[str, str]] = None,
+        merge: bool = True,
     ) -> T:
         """Configures the step.
 
@@ -643,14 +643,14 @@ class BaseStep:
             on_success: Callback function in event of success of the step. Can
                 be a function with no arguments, or a source path to such a
                 function (e.g. `module.my_function`).
-            model: configuration of the model version in the Model Control Plane.
+            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.
             merge: If `True`, will merge the given dictionary configurations
                 like `parameters` and `settings` with existing
                 configurations. If `False` the given configurations will
                 overwrite all existing ones. See the general description of this
                 method for an example.
-            retry: Configuration for retrying the step in case of failure.
-            substitutions: Extra placeholders to use in the name template.
 
         Returns:
             The step instance that this method was called on.
@@ -739,8 +739,9 @@ class BaseStep:
         on_failure: Optional["HookSpecification"] = None,
         on_success: Optional["HookSpecification"] = None,
         model: Optional["Model"] = None,
-        merge: bool = True,
+        retry: Optional[StepRetryConfig] = None,
         substitutions: Optional[Dict[str, str]] = None,
+        merge: bool = True,
     ) -> "BaseStep":
         """Copies the step and applies the given configurations.
 
@@ -766,13 +767,14 @@ class BaseStep:
             on_success: Callback function in event of success of the step. Can
                 be a function with no arguments, or a source path to such a
                 function (e.g. `module.my_function`).
-            model: configuration of the model version in the Model Control Plane.
+            model: Model to use for this step.
+            retry: Configuration for retrying the step in case of failure.
+            substitutions: Extra placeholders for the step name.
             merge: If `True`, will merge the given dictionary configurations
                 like `parameters` and `settings` with existing
                 configurations. If `False` the given configurations will
                 overwrite all existing ones. See the general description of this
                 method for an example.
-            substitutions: Extra placeholders for the step name.
 
         Returns:
             The copied step instance.
@@ -792,8 +794,9 @@ class BaseStep:
             on_failure=on_failure,
             on_success=on_success,
             model=model,
-            merge=merge,
+            retry=retry,
             substitutions=substitutions,
+            merge=merge,
         )
         return step_copy
 

zenml/utils/secret_utils.py

@@ -102,7 +102,7 @@ def SecretField(*args: Any, **kwargs: Any) -> Any:
     """
     json_schema_extra = kwargs.get("json_schema_extra", {})
     json_schema_extra.update({PYDANTIC_SENSITIVE_FIELD_MARKER: True})
-    return Field(json_schema_extra=json_schema_extra, *args, **kwargs)  # type: ignore[pydantic-field]
+    return Field(json_schema_extra=json_schema_extra, *args, **kwargs)
 
 
 def ClearTextField(*args: Any, **kwargs: Any) -> Any:
@@ -119,7 +119,7 @@ def ClearTextField(*args: Any, **kwargs: Any) -> Any:
     """
     json_schema_extra = kwargs.get("json_schema_extra", {})
     json_schema_extra.update({PYDANTIC_CLEAR_TEXT_FIELD_MARKER: True})
-    return Field(json_schema_extra=json_schema_extra, *args, **kwargs)  # type: ignore[pydantic-field]
+    return Field(json_schema_extra=json_schema_extra, *args, **kwargs)
 
 
 def is_secret_field(field: "FieldInfo") -> bool:

zenml/zen_server/dashboard/assets/{404-2I8egBQu.js → 404-CYPi9d8E.js}

@@ -1 +1 @@
-import{j as e}from"./@radix-AvWw-1nd.js";import{h as s,r as t}from"./index-DPjvk73v.js";import{E as r}from"./EmptyState-DpbfQBDE.js";import{S as a}from"./help-CfT0tY2I.js";import{L as o}from"./@react-router-BUo5vhN4.js";import"./@tanstack-CcI3lvwB.js";import"./@reactflow-BHoFKFSZ.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-C7hRs6Kx.js";import{h as s,r as t}from"./index-Be2jLIYM.js";import{E as r}from"./EmptyState-M1jafpg6.js";import{S as a}from"./help-DyMolRxD.js";import{L as o}from"./@react-router-BGgzhn5D.js";import"./@tanstack-CSxjHCME.js";import"./@reactflow-CQi1Z1Wq.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};