zenml-nightly 0.81.0.dev20250428__py3-none-any.whl → 0.81.0.dev20250429__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.81.0.dev20250428
+0.81.0.dev20250429

zenml/config/build_configuration.py

@@ -95,6 +95,10 @@ class BuildConfiguration(BaseModel):
         for _, requirements, _ in requirements_files:
             hash_.update(requirements.encode())
 
+        if self.settings.dockerfile:
+            with open(self.settings.dockerfile, "rb") as f:
+                hash_.update(f.read())
+
         return hash_.hexdigest()
 
     def should_include_files(

zenml/entrypoints/base_entrypoint_configuration.py

@@ -274,7 +274,7 @@ class BaseEntrypointConfiguration(ABC):
         download_dir = os.path.join(
             code_repo_root, code_reference.subdirectory
         )
-        os.makedirs(download_dir)
+        os.makedirs(download_dir, exist_ok=True)
         repo.download_files(
             commit=code_reference.commit,
             directory=download_dir,

zenml/environment.py

@@ -15,12 +15,10 @@
 
 import os
 import platform
-from pathlib import Path
 from typing import Dict
 
 import distro
 
-from zenml import __version__
 from zenml.constants import INSIDE_ZENML_CONTAINER
 from zenml.enums import EnvironmentType
 from zenml.logger import get_logger
@@ -88,35 +86,8 @@ def get_environment() -> str:
         return EnvironmentType.NATIVE
 
 
-def get_system_details() -> str:
-    """Returns OS, python and ZenML information.
-
-    Returns:
-        str: OS, python and ZenML information
-    """
-    from zenml.integrations.registry import integration_registry
-
-    info = {
-        "ZenML version": __version__,
-        "Install path": Path(__file__).resolve().parent,
-        "Python version": Environment.python_version(),
-        "Platform information": Environment.get_system_info(),
-        "Environment": get_environment(),
-        "Integrations": integration_registry.get_installed_integrations(),
-    }
-    return "\n".join(
-        "{:>10} {}".format(k + ":", str(v).replace("\n", " "))
-        for k, v in info.items()
-    )
-
-
 class Environment(metaclass=SingletonMetaClass):
-    """Provides environment information.
-
-    Individual environment components can be registered separately to extend
-    the global Environment object with additional information (see
-    `BaseEnvironmentComponent`).
-    """
+    """Provides environment information."""
 
     def __init__(self) -> None:
         """Initializes an Environment instance.

zenml/exceptions.py

@@ -70,40 +70,6 @@ class DoesNotExistException(ZenMLBaseException):
         super().__init__(message)
 
 
-class PipelineNotSucceededException(ZenMLBaseException):
-    """Raises exception when trying to fetch artifacts from a not succeeded pipeline."""
-
-    def __init__(
-        self,
-        name: str = "",
-        message: str = "{} is not yet completed successfully.",
-    ):
-        """Initializes the exception.
-
-        Args:
-            name: Name of the pipeline.
-            message: Message with details of exception.
-        """
-        super().__init__(message.format(name))
-
-
-class GitException(ZenMLBaseException):
-    """Raises exception when a problem occurs in git resolution."""
-
-    def __init__(
-        self,
-        message: str = "There is a problem with git resolution. "
-        "Please make sure that all relevant files "
-        "are committed.",
-    ):
-        """Initializes the exception.
-
-        Args:
-            message: Message with details of exception.
-        """
-        super().__init__(message)
-
-
 class StepInterfaceError(ZenMLBaseException):
     """Raises exception when interacting with the Step interface in an unsupported way."""
 
@@ -116,50 +82,18 @@ class StepContextError(ZenMLBaseException):
     """Raises exception when interacting with a StepContext in an unsupported way."""
 
 
-class PipelineInterfaceError(ZenMLBaseException):
-    """Raises exception when interacting with the Pipeline interface in an unsupported way."""
-
-
-class ArtifactInterfaceError(ZenMLBaseException):
-    """Raises exception when interacting with the Artifact interface in an unsupported way."""
-
-
 class StackComponentInterfaceError(ZenMLBaseException):
     """Raises exception when interacting with the stack components in an unsupported way."""
 
 
-class StackComponentDeploymentError(ZenMLBaseException):
-    """Raises exception when deploying a stack component fails."""
-
-
 class ArtifactStoreInterfaceError(ZenMLBaseException):
     """Raises exception when interacting with the Artifact Store interface in an unsupported way."""
 
 
-class PipelineConfigurationError(ZenMLBaseException):
-    """Raises exceptions when a pipeline configuration contains invalid values."""
-
-
 class IntegrationError(ZenMLBaseException):
     """Raises exceptions when a requested integration can not be activated."""
 
 
-class DuplicateRunNameError(RuntimeError):
-    """Raises exception when a run with the same name already exists."""
-
-    def __init__(
-        self,
-        message: str = "Unable to run a pipeline with a run name that "
-        "already exists.",
-    ):
-        """Initializes the exception.
-
-        Args:
-            message: Message with details of exception.
-        """
-        super().__init__(message)
-
-
 class ValidationError(ZenMLBaseException):
     """Raised when the Model passed to the ZenStore."""
 
@@ -180,22 +114,10 @@ class StackValidationError(ZenMLBaseException):
     """Raised when a stack configuration is not valid."""
 
 
-class StackComponentValidationError(ZenMLBaseException):
-    """Raised when a stack component configuration is not valid."""
-
-
-class ProvisioningError(ZenMLBaseException):
-    """Raised when an error occurs when provisioning resources for a StackComponent."""
-
-
 class GitNotFoundError(ImportError):
     """Raised when ZenML CLI is used to interact with examples on a machine with no git installation."""
 
 
-class DuplicatedConfigurationError(ZenMLBaseException):
-    """Raised when a configuration parameter is set twice."""
-
-
 class IllegalOperationError(ZenMLBaseException):
     """Raised when an illegal operation is attempted."""
 

zenml/integrations/aws/__init__.py

@@ -41,7 +41,7 @@ class AWSIntegration(Integration):
 
     NAME = AWS
     REQUIREMENTS = [
-        "sagemaker>=2.199.0",
+        "sagemaker>=2.237.3",
         "kubernetes",
         "aws-profile-manager",
     ]

zenml/integrations/evidently/__init__.py

@@ -56,6 +56,7 @@ class EvidentlyIntegration(Integration):
     REQUIREMENTS = [
         "evidently>=0.4.16,<=0.4.22",
         "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+        "numpy<2.0.0",  # evidently is not compatible with NumPy 2.0
     ]
     REQUIREMENTS_IGNORED_ON_UNINSTALL = ["tenacity", "pandas"]
 

zenml/integrations/gcp/flavors/vertex_orchestrator_flavor.py

@@ -112,6 +112,11 @@ class VertexOrchestratorConfig(
         network: the full name of the Compute Engine Network to which the job
             should be peered. For example, `projects/12345/global/networks/myVPC`
             If not provided, the job will not be peered with any network.
+        private_service_connect: the full name of a Private Service Connect
+            endpoint to which the job should be peered. For example,
+            `projects/12345/regions/us-central1/networkAttachments/NETWORK_ATTACHMENT_NAME`
+            If not provided, the job will not be peered with any private service
+            connect endpoint.
         cpu_limit: The maximum CPU limit for this operator. This string value
             can be a number (integer value for number of CPUs) as string,
             or a number followed by "m", which means 1/1000. You can specify
@@ -130,6 +135,7 @@ class VertexOrchestratorConfig(
     encryption_spec_key_name: Optional[str] = None
     workload_service_account: Optional[str] = None
     network: Optional[str] = None
+    private_service_connect: Optional[str] = None
 
     # Deprecated
     cpu_limit: Optional[str] = None

zenml/integrations/gcp/orchestrators/vertex_orchestrator.py

@@ -48,7 +48,15 @@ from uuid import UUID
 
 from google.api_core import exceptions as google_exceptions
 from google.cloud import aiplatform
+from google.cloud.aiplatform.compat.services import (
+    pipeline_service_client_v1beta1,
+)
+from google.cloud.aiplatform.compat.types import pipeline_job_v1beta1
 from google.cloud.aiplatform_v1.types import PipelineState
+from google.cloud.aiplatform_v1beta1.types.service_networking import (
+    PscInterfaceConfig,
+)
+from google.protobuf import json_format
 from google_cloud_pipeline_components.v1.custom_job.utils import (
     create_custom_training_job_from_component,
 )
@@ -495,36 +503,16 @@ class VertexOrchestrator(ContainerizedOrchestrator, GoogleCredentialsMixin):
                 )
                 pod_settings = step_settings.pod_settings
                 if pod_settings:
-                    if pod_settings.host_ipc:
-                        logger.warning(
-                            "Host IPC is set to `True` but not supported in "
-                            "this orchestrator. Ignoring..."
-                        )
-                    if pod_settings.affinity:
-                        logger.warning(
-                            "Affinity is set but not supported in Vertex with "
-                            "Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.tolerations:
-                        logger.warning(
-                            "Tolerations are set but not supported in "
-                            "Vertex with Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.volumes:
-                        logger.warning(
-                            "Volumes are set but not supported in Vertex with "
-                            "Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.volume_mounts:
-                        logger.warning(
-                            "Volume mounts are set but not supported in "
-                            "Vertex with Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.env or pod_settings.env_from:
+                    ignored_fields = pod_settings.model_fields_set - {
+                        "node_selectors"
+                    }
+                    if ignored_fields:
                         logger.warning(
-                            "Environment variables are set but not supported "
-                            "in Vertex with Vertex Pipelines 2.x. Ignoring..."
+                            f"The following pod settings are not supported in "
+                            f"Vertex with Vertex Pipelines 2.x and will be "
+                            f"ignored: {list(ignored_fields)}."
                         )
+
                     for key in pod_settings.node_selectors:
                         if (
                             key
@@ -737,6 +725,39 @@ class VertexOrchestrator(ContainerizedOrchestrator, GoogleCredentialsMixin):
                 self.config.network,
             )
 
+        if self.config.private_service_connect:
+            # The PSC setting isn't yet part of the stable v1 API. We need to
+            # temporarily hack the aiplatform.PipelineJob object in two places:
+            # * to use the v1beta1 PipelineJob primitive which supports the
+            #   psc_interface_config field instead of the v1 PipelineJob
+            #   primitive.
+            # * to use the v1beta1 PipelineServiceClient instead of the v1
+            #   PipelineServiceClient.
+            #
+            # We achieve the first by converting the v1 PipelineJob to a
+            # v1beta1 PipelineJob and the second by replacing the v1
+            # PipelineServiceClient with a v1beta1 PipelineServiceClient.
+            #
+            # TODO: Remove this once the v1 stable API is updated to support
+            # the PSC setting.
+            pipeline_job_dict = json_format.MessageToDict(
+                run._gca_resource._pb, preserving_proto_field_name=True
+            )
+            run._gca_resource = pipeline_job_v1beta1.PipelineJob(
+                **pipeline_job_dict,
+                psc_interface_config=PscInterfaceConfig(
+                    network_attachment=self.config.private_service_connect,
+                ),
+            )
+            run.api_client = (
+                pipeline_service_client_v1beta1.PipelineServiceClient(
+                    credentials=run.credentials,
+                    client_options={
+                        "api_endpoint": run.api_client.api_endpoint,
+                    },
+                )
+            )
+
         try:
             if schedule:
                 logger.info(

zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py

@@ -555,36 +555,14 @@ class KubeflowOrchestrator(ContainerizedOrchestrator):
                 )
                 pod_settings = step_settings.pod_settings
                 if pod_settings:
-                    if pod_settings.host_ipc:
+                    ignored_fields = pod_settings.model_fields_set - {
+                        "node_selectors"
+                    }
+                    if ignored_fields:
                         logger.warning(
-                            "Host IPC is set to `True` but not supported in "
-                            "this orchestrator. Ignoring..."
-                        )
-                    if pod_settings.affinity:
-                        logger.warning(
-                            "Affinity is set but not supported in Kubeflow with "
-                            "Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.tolerations:
-                        logger.warning(
-                            "Tolerations are set but not supported in "
-                            "Kubeflow with Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.volumes:
-                        logger.warning(
-                            "Volumes are set but not supported in Kubeflow with "
-                            "Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.volume_mounts:
-                        logger.warning(
-                            "Volume mounts are set but not supported in "
-                            "Kubeflow with Kubeflow Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.env or pod_settings.env_from:
-                        logger.warning(
-                            "Environment variables are set but not supported "
-                            "in Kubeflow with Kubeflow Pipelines 2.x. "
-                            "Ignoring..."
+                            f"The following pod settings are not supported in "
+                            f"Kubeflow with Kubeflow Pipelines 2.x and will be "
+                            f"ignored: {list(ignored_fields)}."
                         )
 
                     # apply pod settings

zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py

@@ -15,6 +15,8 @@
 
 from typing import TYPE_CHECKING, Optional, Type
 
+from pydantic import PositiveInt
+
 from zenml.config.base_settings import BaseSettings
 from zenml.constants import KUBERNETES_CLUSTER_RESOURCE_TYPE
 from zenml.integrations.kubernetes import KUBERNETES_ORCHESTRATOR_FLAVOR
@@ -47,6 +49,7 @@ class KubernetesOrchestratorSettings(BaseSettings):
         pod_settings: Pod settings to apply to pods executing the steps.
         orchestrator_pod_settings: Pod settings to apply to the pod which is
             launching the actual steps.
+        pod_name_prefix: Prefix to use for the pod name.
         pod_startup_timeout: The maximum time to wait for a pending step pod to
             start (in seconds).
         pod_failure_max_retries: The maximum number of times to retry a step
@@ -55,6 +58,7 @@ class KubernetesOrchestratorSettings(BaseSettings):
             failure retries and pod startup retries (in seconds)
         pod_failure_backoff: The backoff factor for pod failure retries and
             pod startup retries.
+        max_parallelism: Maximum number of steps to run in parallel.
     """
 
     synchronous: bool = True
@@ -64,10 +68,12 @@ class KubernetesOrchestratorSettings(BaseSettings):
     privileged: bool = False
     pod_settings: Optional[KubernetesPodSettings] = None
     orchestrator_pod_settings: Optional[KubernetesPodSettings] = None
+    pod_name_prefix: Optional[str] = None
     pod_startup_timeout: int = 60 * 10  # Default 10 minutes
     pod_failure_max_retries: int = 3
     pod_failure_retry_delay: int = 10
     pod_failure_backoff: float = 1.0
+    max_parallelism: Optional[PositiveInt] = None
 
 
 class KubernetesOrchestratorConfig(

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py

@@ -417,6 +417,9 @@ class KubernetesOrchestrator(ContainerizedOrchestrator):
                 )
 
         pipeline_name = deployment.pipeline_configuration.name
+        settings = cast(
+            KubernetesOrchestratorSettings, self.get_settings(deployment)
+        )
 
         # We already make sure the orchestrator run name has the correct length
         # to make sure we don't cut off the randomized suffix later when
@@ -427,8 +430,16 @@ class KubernetesOrchestrator(ContainerizedOrchestrator):
         orchestrator_run_name = get_orchestrator_run_name(
             pipeline_name, max_length=max_length
         )
+
+        if settings.pod_name_prefix:
+            pod_name = get_orchestrator_run_name(
+                settings.pod_name_prefix, max_length=max_length
+            )
+        else:
+            pod_name = orchestrator_run_name
+
         pod_name = kube_utils.sanitize_pod_name(
-            orchestrator_run_name, namespace=self.config.kubernetes_namespace
+            pod_name, namespace=self.config.kubernetes_namespace
         )
 
         assert stack.container_registry
@@ -455,10 +466,6 @@ class KubernetesOrchestrator(ContainerizedOrchestrator):
             run_id=placeholder_run.id if placeholder_run else None,
         )
 
-        settings = cast(
-            KubernetesOrchestratorSettings, self.get_settings(deployment)
-        )
-
         # Authorize pod to run Kubernetes commands inside the cluster.
         service_account_name = self._get_service_account_name(settings)
 

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py

@@ -15,7 +15,7 @@
 
 import argparse
 import socket
-from typing import Any, Dict
+from typing import Any, Dict, cast
 from uuid import UUID
 
 from kubernetes import client as k8s_client
@@ -41,7 +41,10 @@ from zenml.integrations.kubernetes.orchestrators.manifest_utils import (
 from zenml.logger import get_logger
 from zenml.orchestrators import publish_utils
 from zenml.orchestrators.dag_runner import NodeStatus, ThreadedDagRunner
-from zenml.orchestrators.utils import get_config_environment_vars
+from zenml.orchestrators.utils import (
+    get_config_environment_vars,
+    get_orchestrator_run_name,
+)
 
 logger = get_logger(__name__)
 
@@ -103,8 +106,27 @@ def main() -> None:
         Raises:
             Exception: If the pod fails to start.
         """
-        # Define Kubernetes pod name.
-        pod_name = f"{orchestrator_run_id}-{step_name}"
+        step_config = deployment_config.step_configurations[step_name].config
+        settings = step_config.settings.get("orchestrator.kubernetes", None)
+        settings = KubernetesOrchestratorSettings.model_validate(
+            settings.model_dump() if settings else {}
+        )
+
+        if settings.pod_name_prefix and not orchestrator_run_id.startswith(
+            settings.pod_name_prefix
+        ):
+            max_length = (
+                kube_utils.calculate_max_pod_name_length_for_namespace(
+                    namespace=args.kubernetes_namespace
+                )
+            )
+            pod_name_prefix = get_orchestrator_run_name(
+                settings.pod_name_prefix, max_length=max_length
+            )
+            pod_name = f"{pod_name_prefix}-{step_name}"
+        else:
+            pod_name = f"{orchestrator_run_id}-{step_name}"
+
         pod_name = kube_utils.sanitize_pod_name(
             pod_name, namespace=args.kubernetes_namespace
         )
@@ -116,20 +138,6 @@ def main() -> None:
             step_name=step_name, deployment_id=deployment_config.id
         )
 
-        step_config = deployment_config.step_configurations[step_name].config
-
-        kubernetes_settings = step_config.settings.get(
-            "orchestrator.kubernetes", None
-        )
-
-        orchestrator_settings = {}
-        if kubernetes_settings is not None:
-            orchestrator_settings = kubernetes_settings.model_dump()
-
-        settings = KubernetesOrchestratorSettings.model_validate(
-            orchestrator_settings
-        )
-
         # We set some default minimum memory resource requests for the step pod
         # here if the user has not specified any, because the step pod takes up
         # some memory resources itself and, if not specified, the pod will be
@@ -274,12 +282,17 @@ def main() -> None:
     parallel_node_startup_waiting_period = (
         orchestrator.config.parallel_step_startup_waiting_period or 0.0
     )
+    settings = cast(
+        KubernetesOrchestratorSettings,
+        orchestrator.get_settings(deployment_config),
+    )
     try:
         ThreadedDagRunner(
             dag=pipeline_dag,
             run_fn=run_step_on_kubernetes,
             finalize_fn=finalize_run,
             parallel_node_startup_waiting_period=parallel_node_startup_waiting_period,
+            max_parallelism=settings.max_parallelism,
         ).run()
         logger.info("Orchestration pod completed.")
     finally:

zenml/integrations/kubernetes/orchestrators/manifest_utils.py

@@ -28,6 +28,9 @@ from zenml.integrations.airflow.orchestrators.dag_generator import (
 )
 from zenml.integrations.kubernetes.orchestrators import kube_utils
 from zenml.integrations.kubernetes.pod_settings import KubernetesPodSettings
+from zenml.logger import get_logger
+
+logger = get_logger(__name__)
 
 
 def add_local_stores_mount(
@@ -244,6 +247,24 @@ def add_pod_settings(
     if settings.host_ipc:
         pod_spec.host_ipc = settings.host_ipc
 
+    if settings.scheduler_name:
+        pod_spec.scheduler_name = settings.scheduler_name
+
+    for key, value in settings.additional_pod_spec_args.items():
+        if not hasattr(pod_spec, key):
+            logger.warning(f"Ignoring invalid Pod Spec argument `{key}`.")
+        else:
+            if value is None:
+                continue
+
+            existing_value = getattr(pod_spec, key)
+            if isinstance(existing_value, list):
+                existing_value.extend(value)
+            elif isinstance(existing_value, dict):
+                existing_value.update(value)
+            else:
+                setattr(pod_spec, key, value)
+
 
 def build_cron_job_manifest(
     cron_expression: str,

zenml/integrations/kubernetes/pod_settings.py

@@ -13,7 +13,7 @@
 #  permissions and limitations under the License.
 """Kubernetes pod settings."""
 
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional
 
 from pydantic import field_validator
 
@@ -33,10 +33,13 @@ class KubernetesPodSettings(BaseSettings):
         volumes: Volumes to mount in the pod.
         volume_mounts: Volume mounts to apply to the pod containers.
         host_ipc: Whether to enable host IPC for the pod.
+        scheduler_name: The name of the scheduler to use for the pod.
         image_pull_secrets: Image pull secrets to use for the pod.
         labels: Labels to apply to the pod.
         env: Environment variables to apply to the container.
         env_from: Environment variables to apply to the container.
+        additional_pod_spec_args: Additional arguments to pass to the pod. These
+            will be applied to the pod spec.
     """
 
     node_selectors: Dict[str, str] = {}
@@ -47,10 +50,12 @@ class KubernetesPodSettings(BaseSettings):
     volumes: List[Dict[str, Any]] = []
     volume_mounts: List[Dict[str, Any]] = []
     host_ipc: bool = False
+    scheduler_name: Optional[str] = None
     image_pull_secrets: List[str] = []
     labels: Dict[str, str] = {}
     env: List[Dict[str, Any]] = []
     env_from: List[Dict[str, Any]] = []
+    additional_pod_spec_args: Dict[str, Any] = {}
 
     @field_validator("volumes", mode="before")
     @classmethod

zenml/integrations/numpy/__init__.py

@@ -21,7 +21,10 @@ class NumpyIntegration(Integration):
     """Definition of Numpy integration for ZenML."""
 
     NAME = NUMPY
-    REQUIREMENTS = ["numpy<2.0.0"]
+    # Support both NumPy 1.x and 2.x
+    # Note: Some other integrations (like evidently) may require NumPy <2.0
+    # until they are updated for NumPy 2.0 compatibility
+    REQUIREMENTS = ["numpy<3.0"]
 
     @classmethod
     def activate(cls) -> None:

zenml/integrations/numpy/materializers/numpy_materializer.py

@@ -15,7 +15,16 @@
 
 import os
 from collections import Counter
-from typing import TYPE_CHECKING, Any, ClassVar, Dict, Tuple, Type
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    ClassVar,
+    Dict,
+    Optional,
+    Tuple,
+    Type,
+    Union,
+)
 
 import numpy as np
 
@@ -36,6 +45,48 @@ DATA_FILENAME = "data.parquet"
 SHAPE_FILENAME = "shape.json"
 DATA_VAR = "data_var"
 
+# Check NumPy version for compatibility handling
+IS_NUMPY_2 = np.lib.NumpyVersion(np.__version__) >= "2.0.0"
+
+# In NumPy 2.0, np.object_ is deprecated in favor of object
+# Let's use the right type based on the NumPy version
+NUMPY_OBJECT_TYPE = object if IS_NUMPY_2 else np.object_
+
+
+def _ensure_dtype_compatibility(arr: "NDArray[Any]") -> "NDArray[Any]":
+    """Ensure consistent dtype handling across NumPy versions.
+
+    Args:
+        arr: NumPy array to ensure compatible dtype handling
+
+    Returns:
+        NumPy array with consistent dtype behavior
+    """
+    if IS_NUMPY_2:
+        return arr  # NumPy 2.0 already preserves precision
+    else:
+        # For 1.x, explicitly preserve precision when needed
+        return arr.astype(arr.dtype, copy=False)
+
+
+def _create_array(
+    data: Any, dtype: Optional[Union["np.dtype[Any]", Type[Any]]] = None
+) -> "NDArray[Any]":
+    """Create arrays with consistent behavior across NumPy versions.
+
+    Args:
+        data: Data to convert to array
+        dtype: Optional dtype to use
+
+    Returns:
+        NumPy array with consistent creation behavior
+    """
+    if IS_NUMPY_2:
+        return np.asarray(data, dtype=dtype)
+    else:
+        # In NumPy 1.x, copy behavior is different
+        return np.array(data, dtype=dtype, copy=False)
+
 
 class NumpyMaterializer(BaseMaterializer):
     """Materializer to read data to and from pandas."""
@@ -60,7 +111,9 @@ class NumpyMaterializer(BaseMaterializer):
 
         if self.artifact_store.exists(numpy_file):
             with self.artifact_store.open(numpy_file, "rb") as f:
-                return np.load(f, allow_pickle=True)
+                arr = np.load(f, allow_pickle=True)
+                # Ensure consistent dtype handling
+                return _ensure_dtype_compatibility(arr)
         elif self.artifact_store.exists(os.path.join(self.uri, DATA_FILENAME)):
             logger.warning(
                 "A legacy artifact was found. "
@@ -86,7 +139,9 @@ class NumpyMaterializer(BaseMaterializer):
                     input_stream = pa.input_stream(f)
                     data = pq.read_table(input_stream)
                 vals = getattr(data.to_pandas(), DATA_VAR).values
-                return np.reshape(vals, shape_tuple)
+                arr = np.reshape(vals, shape_tuple)
+                # Ensure consistent dtype handling
+                return _ensure_dtype_compatibility(arr)
             except ImportError:
                 raise ImportError(
                     "You have an old version of a `NumpyMaterializer` ",
@@ -101,6 +156,9 @@ class NumpyMaterializer(BaseMaterializer):
         Args:
             arr: The numpy array to write.
         """
+        # Ensure consistent dtype handling before saving
+        arr = _ensure_dtype_compatibility(arr)
+
         with self.artifact_store.open(
             os.path.join(self.uri, NUMPY_FILENAME), "wb"
         ) as f:
@@ -186,8 +244,8 @@ class NumpyMaterializer(BaseMaterializer):
         """
         if np.issubdtype(arr.dtype, np.number):
             return self._extract_numeric_metadata(arr)
-        elif np.issubdtype(arr.dtype, np.unicode_) or np.issubdtype(
-            arr.dtype, np.object_
+        elif np.issubdtype(arr.dtype, np.str_) or np.issubdtype(
+            arr.dtype, NUMPY_OBJECT_TYPE
         ):
             return self._extract_text_metadata(arr)
         else:
@@ -204,6 +262,9 @@ class NumpyMaterializer(BaseMaterializer):
         Returns:
             A dictionary of metadata.
         """
+        # Ensure consistent precision handling
+        arr = _ensure_dtype_compatibility(arr)
+
         min_val = np.min(arr).item()
         max_val = np.max(arr).item()
 
@@ -228,7 +289,15 @@ class NumpyMaterializer(BaseMaterializer):
         Returns:
             A dictionary of metadata.
         """
-        text = " ".join(arr)
+        # Convert all array elements to strings explicitly to handle
+        # mixed types and ensure NumPy 2.0 compatibility
+        str_items = [str(item) for item in arr.flat]
+        # Use dtype='U' (unicode string) instead of str to avoid type issues
+        str_arr = _create_array(str_items, dtype=np.dtype("U")).reshape(
+            arr.shape
+        )
+
+        text = " ".join(str_arr)
         words = text.split()
         word_counts = Counter(words)
         unique_words = len(word_counts)

zenml/integrations/tekton/orchestrators/tekton_orchestrator.py

@@ -524,36 +524,16 @@ class TektonOrchestrator(ContainerizedOrchestrator):
                 node_selector_constraint: Optional[Tuple[str, str]] = None
                 pod_settings = step_settings.pod_settings
                 if pod_settings:
-                    if pod_settings.host_ipc:
+                    ignored_fields = pod_settings.model_fields_set - {
+                        "node_selectors"
+                    }
+                    if ignored_fields:
                         logger.warning(
-                            "Host IPC is set to `True` but not supported in "
-                            "this orchestrator. Ignoring..."
-                        )
-                    if pod_settings.affinity:
-                        logger.warning(
-                            "Affinity is set but not supported in Tekton with "
-                            "Tekton Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.tolerations:
-                        logger.warning(
-                            "Tolerations are set but not supported in "
-                            "Tekton with Tekton Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.volumes:
-                        logger.warning(
-                            "Volumes are set but not supported in Tekton with "
-                            "Tekton Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.volume_mounts:
-                        logger.warning(
-                            "Volume mounts are set but not supported in "
-                            "Tekton with Tekton Pipelines 2.x. Ignoring..."
-                        )
-                    if pod_settings.env or pod_settings.env_from:
-                        logger.warning(
-                            "Environment variables are set but not supported "
-                            "in Tekton with Tekton Pipelines 2.x. Ignoring..."
+                            f"The following pod settings are not supported in "
+                            f"Tekton with Tekton Pipelines 2.x and will be "
+                            f"ignored: {list(ignored_fields)}."
                         )
+
                     # apply pod settings
                     if (
                         KFP_ACCELERATOR_NODE_SELECTOR_CONSTRAINT_LABEL

zenml/orchestrators/dag_runner.py

@@ -51,10 +51,11 @@ def reverse_dag(dag: Dict[str, List[str]]) -> Dict[str, List[str]]:
 class NodeStatus(Enum):
     """Status of the execution of a node."""
 
-    WAITING = "Waiting"
-    RUNNING = "Running"
-    COMPLETED = "Completed"
-    FAILED = "Failed"
+    NOT_STARTED = "not_started"
+    PENDING = "pending"
+    RUNNING = "running"
+    COMPLETED = "completed"
+    FAILED = "failed"
 
 
 class ThreadedDagRunner:
@@ -73,6 +74,7 @@ class ThreadedDagRunner:
         run_fn: Callable[[str], Any],
         finalize_fn: Optional[Callable[[Dict[str, NodeStatus]], None]] = None,
         parallel_node_startup_waiting_period: float = 0.0,
+        max_parallelism: Optional[int] = None,
     ) -> None:
         """Define attributes and initialize all nodes in waiting state.
 
@@ -85,16 +87,26 @@ class ThreadedDagRunner:
                 when all nodes have completed.
             parallel_node_startup_waiting_period: Delay in seconds to wait in
                 between starting parallel nodes.
+            max_parallelism: Maximum number of nodes to run in parallel
+
+        Raises:
+            ValueError: If max_parallelism is not greater than 0.
         """
+        if max_parallelism is not None and max_parallelism <= 0:
+            raise ValueError("max_parallelism must be greater than 0")
+
         self.parallel_node_startup_waiting_period = (
             parallel_node_startup_waiting_period
         )
+        self.max_parallelism = max_parallelism
         self.dag = dag
         self.reversed_dag = reverse_dag(dag)
         self.run_fn = run_fn
         self.finalize_fn = finalize_fn
         self.nodes = dag.keys()
-        self.node_states = {node: NodeStatus.WAITING for node in self.nodes}
+        self.node_states = {
+            node: NodeStatus.NOT_STARTED for node in self.nodes
+        }
         self._lock = threading.Lock()
 
     def _can_run(self, node: str) -> bool:
@@ -109,8 +121,7 @@ class ThreadedDagRunner:
         Returns:
             True if the node can run else False.
         """
-        # Check that node has not run yet.
-        if not self.node_states[node] == NodeStatus.WAITING:
+        if not self.node_states[node] == NodeStatus.NOT_STARTED:
             return False
 
         # Check that all upstream nodes of this node have already completed.
@@ -120,6 +131,33 @@ class ThreadedDagRunner:
 
         return True
 
+    def _prepare_node_run(self, node: str) -> None:
+        """Prepare a node run.
+
+        Args:
+            node: The node.
+        """
+        if self.max_parallelism is None:
+            with self._lock:
+                self.node_states[node] = NodeStatus.RUNNING
+        else:
+            while True:
+                with self._lock:
+                    logger.debug(f"Checking if {node} can run.")
+                    running_nodes = len(
+                        [
+                            state
+                            for state in self.node_states.values()
+                            if state == NodeStatus.RUNNING
+                        ]
+                    )
+                    if running_nodes < self.max_parallelism:
+                        self.node_states[node] = NodeStatus.RUNNING
+                        break
+
+                logger.debug(f"Waiting for {running_nodes} nodes to finish.")
+                time.sleep(10)
+
     def _run_node(self, node: str) -> None:
         """Run a single node.
 
@@ -128,6 +166,8 @@ class ThreadedDagRunner:
         Args:
             node: The node.
         """
+        self._prepare_node_run(node)
+
         try:
             self.run_fn(node)
             self._finish_node(node)
@@ -138,19 +178,15 @@ class ThreadedDagRunner:
     def _run_node_in_thread(self, node: str) -> threading.Thread:
         """Run a single node in a separate thread.
 
-        First updates the node status to running.
-        Then calls self._run_node() in a new thread and returns the thread.
-
         Args:
             node: The node.
 
         Returns:
             The thread in which the node was run.
         """
-        # Update node status to running.
-        assert self.node_states[node] == NodeStatus.WAITING
+        assert self.node_states[node] == NodeStatus.NOT_STARTED
         with self._lock:
-            self.node_states[node] = NodeStatus.RUNNING
+            self.node_states[node] = NodeStatus.PENDING
 
         # Run node in new thread.
         thread = threading.Thread(target=self._run_node, args=(node,))
@@ -225,7 +261,7 @@ class ThreadedDagRunner:
         for node in self.nodes:
             if self.node_states[node] == NodeStatus.FAILED:
                 failed_nodes.append(node)
-            elif self.node_states[node] == NodeStatus.WAITING:
+            elif self.node_states[node] == NodeStatus.NOT_STARTED:
                 skipped_nodes.append(node)
 
         if failed_nodes:

zenml/utils/code_utils.py

@@ -283,7 +283,7 @@ def download_code_from_artifact_store(
     artifact_store._register()
 
     extract_dir = os.path.abspath("code")
-    os.makedirs(extract_dir)
+    os.makedirs(extract_dir, exist_ok=True)
 
     download_and_extract_code(code_path=code_path, extract_dir=extract_dir)
 

zenml/zen_server/exceptions.py

@@ -22,7 +22,6 @@ from zenml.exceptions import (
     AuthorizationException,
     CredentialsNotValid,
     DoesNotExistException,
-    DuplicateRunNameError,
     EntityCreationError,
     EntityExistsError,
     IllegalOperationError,
@@ -69,7 +68,6 @@ error_response = dict(model=ErrorModel)
 # different status codes (e.g. `ValueError` and the 400 and 422 status codes).
 REST_API_EXCEPTIONS: List[Tuple[Type[Exception], int]] = [
     # 409 Conflict
-    (DuplicateRunNameError, 409),
     (EntityExistsError, 409),
     # 403 Forbidden
     (IllegalOperationError, 403),

{zenml_nightly-0.81.0.dev20250428.dist-info → zenml_nightly-0.81.0.dev20250429.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.81.0.dev20250428
+Version: 0.81.0.dev20250429
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -110,7 +110,7 @@ Requires-Dist: requests (>=2.27.11,<3.0.0) ; extra == "connectors-azure"
 Requires-Dist: rich[jupyter] (>=12.0.0)
 Requires-Dist: ruff (>=0.1.7) ; extra == "templates" or extra == "dev"
 Requires-Dist: s3fs (>=2022.11.0,!=2025.3.1) ; extra == "s3fs"
-Requires-Dist: sagemaker (>=2.199.0) ; extra == "sagemaker"
+Requires-Dist: sagemaker (>=2.237.3) ; extra == "sagemaker"
 Requires-Dist: secure (>=0.3.0,<0.4.0) ; extra == "server"
 Requires-Dist: setuptools
 Requires-Dist: sqlalchemy (>=2.0.0,<3.0.0)

{zenml_nightly-0.81.0.dev20250428.dist-info → zenml_nightly-0.81.0.dev20250429.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=mqIIfKSgM9a747yV1iWctiuoWQks6eRjizzoouMuDDg,19
+zenml/VERSION,sha256=fikJ9F_ekw4wug-gVMiZuNsv_TnkQEfUs2JTtk2Acvw,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -64,7 +64,7 @@ zenml/code_repositories/git/local_git_repository_context.py,sha256=hfX7zVDQ27Le0
 zenml/code_repositories/local_repository_context.py,sha256=1VyiYkJBDVg0iGusgRQDToGRPJuu9lx7jTBDpplukDg,2816
 zenml/config/__init__.py,sha256=DZEic7euSbwI9Yb3FMRQhTgfhqz-C6OdAiYmOb0-opI,1519
 zenml/config/base_settings.py,sha256=itoLqc1cOwEYhgSGdZmSKSaBevQkvYH7NQh7PUamazc,1700
-zenml/config/build_configuration.py,sha256=Jsng7ebpeaRbzXlbszU-uYkaVihgQ4OrD839yWwD3ZY,5126
+zenml/config/build_configuration.py,sha256=vpea4cXTWfxRyijMFMo3U6jtHpC7wgL01NAIzp721RI,5263
 zenml/config/compiler.py,sha256=jzyyTmM8nTxtpIVTSjZ-jrVOBJ1roVoJ3fs1A1nwv9M,24130
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=KTUR9FaRn3ivcagzFgpLe4eev8fvd5Wq1oV8wGNMB-g,13322
@@ -96,12 +96,12 @@ zenml/container_registries/github_container_registry.py,sha256=rbcGkFfPDk-Il0T9F
 zenml/data_validators/__init__.py,sha256=9Fa0jiUSQ_JsLMHYjqDayWQl4m_uuai9tQjIP60OTk8,886
 zenml/data_validators/base_data_validator.py,sha256=reGUJ6NEFfd_wocrcYBSU5QHCSBOll-pP_suTP_bfv0,9839
 zenml/entrypoints/__init__.py,sha256=2CMemOrHIJauxAss6k7dKFtsCFqgYR-JbAx4REoCaE8,946
-zenml/entrypoints/base_entrypoint_configuration.py,sha256=5GY0W-29pabMmtb8WlnEKtg3H-ywFEpXzpud3hC12_o,12516
+zenml/entrypoints/base_entrypoint_configuration.py,sha256=t7sU2MEEPVomX9dZCpPhk1f3SVjcN7whQRLsY2uGXu0,12531
 zenml/entrypoints/entrypoint.py,sha256=XNgXBCMKoidmP0_AYgMpqo-neG8Y8jG0rj43ofTDZ9E,2033
 zenml/entrypoints/pipeline_entrypoint_configuration.py,sha256=To-vTP29qAE36ndJDF1fRw9wL2Nk2bsBuO-ayAwvSmo,1646
 zenml/entrypoints/step_entrypoint_configuration.py,sha256=fJuTvJnGuhKc60CH1VMQL5EHomGXkYZulv6pVgd9M6w,7316
 zenml/enums.py,sha256=m18yol20jS9JNhuf7UPBYUsgNRCtbyCaWPNpeEAJeXY,11199
-zenml/environment.py,sha256=dw9sU28d8vsobGl2byUyhVbYXVhgtvpRrEAx4_fHMI8,11624
+zenml/environment.py,sha256=VUS8aPTNa1c5VhXIyeL6VaFTQA7K5Q3ltLu24IYP2oI,10694
 zenml/event_hub/__init__.py,sha256=-fD9mPOslf4J-_XFBPp5gYlPz7-6ZaAKHa5jxf_nyAo,757
 zenml/event_hub/base_event_hub.py,sha256=PqKrnvOye0UUS3s09zGgjI5dtj0IwzEBDbavA_PgfZQ,6579
 zenml/event_hub/event_hub.py,sha256=e1eCRB1qAabOFIx2OJCAoNqN1QQIW0CSTukiWuGMOi8,6502
@@ -110,7 +110,7 @@ zenml/event_sources/base_event.py,sha256=irrpiYi4fDYewzaXtb6_gPsqyFlYYLSap2BTGQz
 zenml/event_sources/base_event_source.py,sha256=04ramQhcPhGD9Mo9BcJZR-b9fJb8a4wzvlPVMAb9Dyk,25893
 zenml/event_sources/webhooks/__init__.py,sha256=VsHzSn1oKFaUs495ZqsKCfXiYxtaeGsnSGuvuqbGXCg,658
 zenml/event_sources/webhooks/base_webhook_event_source.py,sha256=2zADrL3cNpD-akZRdemUePUnTl8pPMPVX6eFQgWeUSo,7371
-zenml/exceptions.py,sha256=-Zt5yOEsj8MkbIGLceLa0JGjQl-PinpQifT0j5bAt2c,9043
+zenml/exceptions.py,sha256=KPEz9kQGyC6VbgjNXFTUsVehvAUIe2ayOOy1j5QJdUI,6757
 zenml/experiment_trackers/__init__.py,sha256=b5XlKoRtMR1WBQVEiItolkpsa0iJ1IqxTmmRatr4YDw,1119
 zenml/experiment_trackers/base_experiment_tracker.py,sha256=K92w7c0r45qLPIsA3YmwPflaIl_WTK8-_Hh_1MQiLkE,2218
 zenml/feature_stores/__init__.py,sha256=tSW7YnDa3NDnlkX3yA_CTdX7ocWB9gsfF-7X9sc6i8Y,1415
@@ -135,7 +135,7 @@ zenml/integrations/argilla/annotators/__init__.py,sha256=QjRMxIQ-skulcLN94GuHuuk
 zenml/integrations/argilla/annotators/argilla_annotator.py,sha256=aIA5rcwfOYLnUVp2c_aaTAUnVW8GV9gP8myze_jY_qY,15586
 zenml/integrations/argilla/flavors/__init__.py,sha256=MTogEeiZ1k7nLzGKok3azK_VhKUAJl0GL6OQvsvMlZo,917
 zenml/integrations/argilla/flavors/argilla_annotator_flavor.py,sha256=NAy6QKLWYXHPTCoWc9RIovsngcRGMwseu3rdNjiCpo4,4473
-zenml/integrations/aws/__init__.py,sha256=48ssbSeI3EUyjZN__Xs350wX5pLKoFdUZpXiXLaDDs4,2452
+zenml/integrations/aws/__init__.py,sha256=WevXkUf7tNGrxkY2wFYPohuBM2Fd40wRE8ff2y0k-ks,2452
 zenml/integrations/aws/container_registries/__init__.py,sha256=fOwo84MiaX_7XhNQEXeyVh8AyOlcIl6nSu_ig68GkSs,827
 zenml/integrations/aws/container_registries/aws_container_registry.py,sha256=SCxKj_w79Ka0LdxYOdDmaXitketTQRDHrQ4XQ0_f4hs,6091
 zenml/integrations/aws/flavors/__init__.py,sha256=XYL9fpwKzeFfHCjakU0iJ3SyHVRJk63QT7luOy9Giek,1480
@@ -230,7 +230,7 @@ zenml/integrations/discord/flavors/discord_alerter_flavor.py,sha256=9hX7R7dfxSwi
 zenml/integrations/discord/steps/__init__.py,sha256=stSDntUMzrHzwMJm1V1-jm7otII7uW6Fxj7qYB7MWrc,663
 zenml/integrations/discord/steps/discord_alerter_ask_step.py,sha256=puBERGjhpBRaift8GCygAgnjgZHbeqclRywxJjjjEG8,2553
 zenml/integrations/discord/steps/discord_alerter_post_step.py,sha256=te4M4Q47e1nShPHLLv414bjDuG_r7XCxDUbLgwGXEtI,2283
-zenml/integrations/evidently/__init__.py,sha256=ZNbkmtHPsk4qs4AColYU9a2VN8oV5XPtUrmb78UTDg0,3185
+zenml/integrations/evidently/__init__.py,sha256=csvwza0td2EQExJCGdZqZXk1NVJY9miyEOszglZPCwU,3254
 zenml/integrations/evidently/column_mapping.py,sha256=slZwGaArhYZNZnXfwYFXZEt7bqq2jswvb1vwkedvGRE,3555
 zenml/integrations/evidently/data_validators/__init__.py,sha256=7H1HCXAefk-asnSAYqfud-l17rsBFfhCrgps2abhmFY,830
 zenml/integrations/evidently/data_validators/evidently_data_validator.py,sha256=V34Nze3Mi4JpTlJJQf-i582WxAZrg5-yAv1HcUf7ULE,10316
@@ -263,13 +263,13 @@ zenml/integrations/gcp/flavors/__init__.py,sha256=GcB8EvYjXM_VSku16jnDSNyJYMgKc2
 zenml/integrations/gcp/flavors/gcp_artifact_store_flavor.py,sha256=Ts2jvR3IgGH8YyaFBT6OQtx2kCKD9dgZgceKiRAv_QI,3560
 zenml/integrations/gcp/flavors/gcp_image_builder_flavor.py,sha256=K6sE9D-okbdlctNwNeDYEfhWmMXXW-S92x342dnhjqY,4451
 zenml/integrations/gcp/flavors/vertex_experiment_tracker_flavor.py,sha256=icexIPoqyJ_tsO6M5-Vncd1TAUaKTGbdUG0cDOYC6Kc,6834
-zenml/integrations/gcp/flavors/vertex_orchestrator_flavor.py,sha256=Cm31d1uNZ6C6E1oQV8ep91IzvboFnyRwpH2ZKojsctg,10096
+zenml/integrations/gcp/flavors/vertex_orchestrator_flavor.py,sha256=dIZP35SR63RHb9QY1bz_yDWrzQ9pz2KKxAorShqUMkg,10494
 zenml/integrations/gcp/flavors/vertex_step_operator_flavor.py,sha256=hY38MoNXZbvylsrbVdQ7PtkNw9QmHj6G8P1BWrRUyv8,5231
 zenml/integrations/gcp/google_credentials_mixin.py,sha256=bPy3JYCCcyuTmPiVFqbY81YJ5g1yRdzHLlBlokvbeqg,4026
 zenml/integrations/gcp/image_builders/__init__.py,sha256=2IvTL6U2YpUoxGQXeXew-6WFoL5hHIxkqr4DaA5Ez9w,786
 zenml/integrations/gcp/image_builders/gcp_image_builder.py,sha256=5T6BXsHxLhvp1BF_rslXl1oZzykJUPuZ3E_7-9ZZYLk,9019
 zenml/integrations/gcp/orchestrators/__init__.py,sha256=6xLFJKZKQk73fHPF-XdpbQO87zjQNGTsNHjJjLfG_Kg,805
-zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=_OCr6NGhBHRuR4t9kOyyvjlSu4EQPC0EY-mrYM56VvU,41354
+zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=qoMCr36buZUz0y4CyTFQde3RDslkaGLAG0FjXc0XEPU,42100
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
 zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=JGrFTKkQV4ZDn_yTEqX-AtBMraFasFgzLVws2mvhS64,94915
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
@@ -330,19 +330,19 @@ zenml/integrations/kubeflow/__init__.py,sha256=4kP4BuSbRs9xXyymv38rgb5l-LJIxb7te
 zenml/integrations/kubeflow/flavors/__init__.py,sha256=l560A0oIpYeTpVVrdrVVYj-2-Y5CeyCSQMfwErZROxY,878
 zenml/integrations/kubeflow/flavors/kubeflow_orchestrator_flavor.py,sha256=b7W4oASHLYN9o2n__-W3zajb_MXhPaY0UutecsW0Xw4,10381
 zenml/integrations/kubeflow/orchestrators/__init__.py,sha256=J879DBt9WbpojBTdOXy7CO6F5OuTMntBeZ8aUY2EGO8,821
-zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=6IxKpOgCmTP0kPnhsGDi3rNGDCb3RnHIFu7DflxOP0Q,39690
+zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=OuTGD7ggsw_tYILOTRT1xz8uiTJNhp0Pkx3Sz_MMFWM,38521
 zenml/integrations/kubeflow/orchestrators/local_deployment_utils.py,sha256=qszoOdvBpgIp40XkncphXAr9dRKnyZzGiz2mJ56bYmw,15448
 zenml/integrations/kubernetes/__init__.py,sha256=k1bfrdI1u5RBnAh7yT4w-m-4SWgZ7b4L5uu7dPRc0SI,1809
 zenml/integrations/kubernetes/flavors/__init__.py,sha256=a5gU45qCj3FkLwl_uVjlIkL_2F5DHk-w1gdcZrvVjBI,1266
-zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=VOfb5yOXGpQ8gcKP0nt4bYO48LqG8ZjzGhpw8XGXAzk,9253
+zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=bO_2RZ6TIO0Qp2zduha324m2zrg6cGw-ymK51Nm2E9o,9505
 zenml/integrations/kubernetes/flavors/kubernetes_step_operator_flavor.py,sha256=xFO7cSusji-mgbRrt4mU29gdyC9iEjEHKtomdFLp9mM,6265
 zenml/integrations/kubernetes/orchestrators/__init__.py,sha256=TJID3OTieZBox36WpQpzD0jdVRA_aZVcs_bNtfXS8ik,811
 zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=oU0EYP-x35oG7PAy7NZKeFA8_89Eckgq6hibwc9v5l0,18108
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=iESgw3Yx7biqt3D05Q1R8u_fhOTuNKpy8H17ntmHuS4,25518
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=aA_1CozChqrAuylzuG8_13JR0eR5HrXCsq_k-Y5PDqc,11499
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=tzGhxFyf2of7soyEiEYAixuU9uDjT49-StAqhRJLMVo,25730
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=krsIXrHUk2EZAVpqMCn8KgICNhD_YtVlySarQD5C4bI,12039
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=KjHfQK9VQEQkkkM2i9w51AzqolgIU01M5dgb2YGamvY,2754
-zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=Owyuz5Iix-QvCnKObC6xhcuQtNG_ik-8Vbdmk13eWfc,12557
-zenml/integrations/kubernetes/pod_settings.py,sha256=fP2NeHf1XxT__D4g_6oHQf8pkiiamFUFYmfNor2OoP8,6386
+zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=Mte6q_NX0bU_JTivtN7ByQtfQX32iI2z2sElIpQhj7g,13271
+zenml/integrations/kubernetes/pod_settings.py,sha256=QPV8Fq8SHZ7pwKavplcADJep-Y9olIbC3RlG10HI0pw,6683
 zenml/integrations/kubernetes/serialization_utils.py,sha256=cPSe4szdBLzDnUZT9nQc2CCA8h84aj5oTA8vsUE36ig,7000
 zenml/integrations/kubernetes/service_connectors/__init__.py,sha256=Uf6zlHIapYrRDl3xOPWQ2jA7jt85SXx1U7DmSxzxTvQ,818
 zenml/integrations/kubernetes/service_connectors/kubernetes_service_connector.py,sha256=Cv4tiVxoQOz9ex0lf3JdJrooEkgMwfDfwt5GOeNRpQU,19669
@@ -411,9 +411,9 @@ zenml/integrations/neptune/neptune_constants.py,sha256=-VddhrALS1HMBhGtFiGDKaRah
 zenml/integrations/neural_prophet/__init__.py,sha256=mVou7yCAImlDA6Qtyh1jhBfL51oFhgl96Sj9dn7PwG4,1309
 zenml/integrations/neural_prophet/materializers/__init__.py,sha256=7O-qwxLjxLmvqeUgeLWBOHe5pvxwDFRg5hCOhffyd0k,802
 zenml/integrations/neural_prophet/materializers/neural_prophet_materializer.py,sha256=U8OvRObumGAk66UJFHLKWtez-I3wU2zd1j-XvCbSBr4,1570
-zenml/integrations/numpy/__init__.py,sha256=iElEW7S10JEHnW50vbqTs5dhm4bSIiB70cJSx38Lnco,1066
+zenml/integrations/numpy/__init__.py,sha256=2VlwCUAne0MYpYGPkgk19yq2gntUCe9IMytLrjY9KEw,1234
 zenml/integrations/numpy/materializers/__init__.py,sha256=txwv8We-dLehTWqY-eDYx40njg4Ld8eQrs1O0MZiiIk,766
-zenml/integrations/numpy/materializers/numpy_materializer.py,sha256=rNIcoZkU6JZcqEc6yt3x3yHvmmWnLAfKy74hLxKXbM8,8477
+zenml/integrations/numpy/materializers/numpy_materializer.py,sha256=LMnchlRl6xOBrQcAi_A_jQZtzEkkix2WQl8cU4z--nU,10602
 zenml/integrations/openai/__init__.py,sha256=Qh9ZDYm5Zn6thmEBpW1WxBUaqOChDYa5sdNzHB4zB4U,919
 zenml/integrations/openai/hooks/__init__.py,sha256=8VfiVOyIrjya9G_VK5GPEqq9G5i9w5u4ngf-Oo_oHT4,811
 zenml/integrations/openai/hooks/open_ai_failure_hook.py,sha256=tQe-dUO7_w24ABfN96TZ7Zc2inJMI5u9sdE8gBxrDyM,4702
@@ -531,7 +531,7 @@ zenml/integrations/tekton/__init__.py,sha256=fR4Swt09E3VWCV-5to3zbCOaZcsDPtDs22z
 zenml/integrations/tekton/flavors/__init__.py,sha256=-S5XuwcZKWyGb9tVfl_gEFJj1KSXX6kwIfe-EhE0TmM,864
 zenml/integrations/tekton/flavors/tekton_orchestrator_flavor.py,sha256=XgFgzJUlje9J1o5zwBvs_ycpgQjGdi50KZFA9_tT0vc,8268
 zenml/integrations/tekton/orchestrators/__init__.py,sha256=yCJEM-PCechO4DF_YQeg82y76nwBKeXTy_SIXRWX2DE,811
-zenml/integrations/tekton/orchestrators/tekton_orchestrator.py,sha256=RnoVj5o3JYqfhcIFfzOunXqbVeSMHNWT9VVKKQmqwlE,35735
+zenml/integrations/tekton/orchestrators/tekton_orchestrator.py,sha256=tRhMNhe9a7hVr1HFYMdfH15nB_xrVlJwrLs9R8qbXec,34614
 zenml/integrations/tensorboard/__init__.py,sha256=mrzcQooqOlOYrzGLyZxxn1VPl99ot9Usg75akPC0Uc0,1612
 zenml/integrations/tensorboard/services/__init__.py,sha256=Y-YVPLxJkFSpXNDZSk8gEBISsXJGk_DgGIyVIIFIa40,798
 zenml/integrations/tensorboard/services/tensorboard_service.py,sha256=OgIO0UTndzGqiDz_OJjCzm9zHa0F0P7kjxzrz83U77g,4434
@@ -682,7 +682,7 @@ zenml/orchestrators/__init__.py,sha256=p4Y9Ni7Lp33fZ6UrjgI7qu-rrg8LrlyafCM-K1WC8
 zenml/orchestrators/base_orchestrator.py,sha256=q0N0o7neQEcnsNHOeGUssbqC_MoQLFuBFavMD-W_QZY,12837
 zenml/orchestrators/cache_utils.py,sha256=G9mIZZOa1TixUHZnvwwroMl7a4YCVqvrNVwPCMVNFos,4306
 zenml/orchestrators/containerized_orchestrator.py,sha256=rdebgBW0Bk--JcHcT0NpLkAbyhY0VS5xO1uwWEgkLpA,3230
-zenml/orchestrators/dag_runner.py,sha256=N3gp1K1kZ8aEHPYyxcjpxmfRFFQa8Z0Nm5QuVz_rVYw,8172
+zenml/orchestrators/dag_runner.py,sha256=4oAGc52nHv8HeI_Vj7GH1jzjJom1uwUJ_la9usQoOFY,9404
 zenml/orchestrators/input_utils.py,sha256=GKDKZBjPLUkXn46bPxzhN28SJwRSvYLvlhB9eEKKPgA,7660
 zenml/orchestrators/local/__init__.py,sha256=qlU91hgqGKeycw-Qntrn-iMuoMTaNrq-RgfOFeqwlM4,662
 zenml/orchestrators/local/local_orchestrator.py,sha256=UhgxC68A_6N7Vp1zNixSUr7qb9_S79Gg-ZnjMJcViw0,5668
@@ -762,7 +762,7 @@ zenml/utils/__init__.py,sha256=jaMTbjm8tLYkaRoxlZ0Em4ye_ZHOHKgP2goPTTiYGUQ,797
 zenml/utils/archivable.py,sha256=QuLe1IhyABTrE6Y0hAT3tKjaUCpcze5ffZ_RKoKtJwY,6549
 zenml/utils/callback_registry.py,sha256=QBWdaraLAxBxi8DKbv9X1SUpTKDhhj-XE0JomB2Ax2Y,2411
 zenml/utils/code_repository_utils.py,sha256=-Zw8TvCSPfISB30Gmj0Yryk-RYrpzwCkqH7_c7Hq9ZY,5334
-zenml/utils/code_utils.py,sha256=SsfE4sHcBX3yPyrk3QWZXpXfsTcjycbVIuqVV6zKPDY,11463
+zenml/utils/code_utils.py,sha256=P9o6jRmvfmq-MKBGJ2XazMnMmDEuSvsFBEhqin4eaws,11478
 zenml/utils/cuda_utils.py,sha256=RR21m__Zs-OnI5n-veFUzWniZjwLSbalHE5QV3jK1Hw,1624
 zenml/utils/daemon.py,sha256=GZ7Dx6GLHK04SR50wBxpKYmFhxPBfdLWxJiAWzJC6cM,11863
 zenml/utils/dashboard_utils.py,sha256=V4pas-zgMn3vOpXWZ-yoW11ZLZi__oFQw9jvhiXhK1A,8345
@@ -1011,7 +1011,7 @@ zenml/zen_server/deploy/docker/docker_provider.py,sha256=18pNpxvP8xqbxS_KxvYTEIu
 zenml/zen_server/deploy/docker/docker_zen_server.py,sha256=83-bjRmtlE-WDCavbvL9uTXOn5fMJ4yPDDb2ewSr4Do,7480
 zenml/zen_server/deploy/exceptions.py,sha256=tX0GNnLB_GMkeN7zGNlJRwtlrpZ5Slvyj_unVYVmGxk,1396
 zenml/zen_server/download_utils.py,sha256=g366TqytIWR4L4u9dd0jg-ouSWpDwdAh_-F2zryFzfo,4224
-zenml/zen_server/exceptions.py,sha256=H4F9SbbwCiVLnJTdzpWN016Knt6zvoOVmOWxTgXEAsA,9686
+zenml/zen_server/exceptions.py,sha256=NAXARDHLvfnVIQ87MHce0mfaWCmKhMKgwmw-_X49HxQ,9625
 zenml/zen_server/feature_gate/__init__.py,sha256=yabe4fBY6NSusn-QlKQDLOvXVLERNpcAQgigsyWQIbQ,612
 zenml/zen_server/feature_gate/endpoint_utils.py,sha256=o6sBVlqqlc9KokMaEsRTYeMra7f2a6kCt3FrB-oHhCw,2227
 zenml/zen_server/feature_gate/feature_gate_interface.py,sha256=XCgsqUN5yYCipodHCIDmCHRhyYMyt483Pp5mdpqzwHA,1647
@@ -1316,8 +1316,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=diS6mZwE0A5v2wKf69CDtwCr1Bveot8QSQc-WE44rw4,442103
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.81.0.dev20250428.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.81.0.dev20250428.dist-info/METADATA,sha256=uKUnevy1dcQwb6j6e8g6u7KeEJYGXskQPXiWGvuzIpw,24304
-zenml_nightly-0.81.0.dev20250428.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-zenml_nightly-0.81.0.dev20250428.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.81.0.dev20250428.dist-info/RECORD,,
+zenml_nightly-0.81.0.dev20250429.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.81.0.dev20250429.dist-info/METADATA,sha256=XFMW9u3QQraG74PaFweznntiXitFHQRSV7w-eY_ijLk,24304
+zenml_nightly-0.81.0.dev20250429.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.81.0.dev20250429.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.81.0.dev20250429.dist-info/RECORD,,