zenml-nightly 0.82.0.dev20250513__py3-none-any.whl → 0.82.0.dev20250514__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.82.0.dev20250513
+0.82.0.dev20250514

zenml/config/build_configuration.py

@@ -135,6 +135,9 @@ class BuildConfiguration(BaseModel):
         Returns:
             Whether files should be included in the image.
         """
+        if self.settings.local_project_install_command:
+            return True
+
         if self.should_download_files(code_repository=code_repository):
             return False
 
@@ -153,6 +156,9 @@ class BuildConfiguration(BaseModel):
         Returns:
             Whether files should be downloaded in the image.
         """
+        if self.settings.local_project_install_command:
+            return False
+
         if self.should_download_files_from_code_repository(
             code_repository=code_repository
         ):
@@ -176,6 +182,9 @@ class BuildConfiguration(BaseModel):
         Returns:
             Whether files should be downloaded from the code repository.
         """
+        if self.settings.local_project_install_command:
+            return False
+
         if (
             code_repository
             and self.settings.allow_download_from_code_repository

zenml/config/docker_settings.py

@@ -91,12 +91,21 @@ class DockerSettings(BaseSettings):
     --------------------------------
     Depending on the configuration of this object, requirements will be
     installed in the following order (each step optional):
-    - The packages installed in your local python environment
+    - The packages installed in your local python environment (extracted using
+      `pip freeze`)
     - The packages required by the stack unless this is disabled by setting
-      `install_stack_requirements=False`.
+      `install_stack_requirements=False`
     - The packages specified via the `required_integrations`
+    - The packages defined inside a pyproject.toml file given by the
+      `pyproject_path` attribute.
     - The packages specified via the `requirements` attribute
 
+    If neither `replicate_local_python_environment`, `pyproject_path` or
+    `requirements` are specified, ZenML will try to automatically find a
+    requirements.txt or pyproject.toml file in your current source root
+    and installs packages from the first one it finds. You can disable this
+    behavior by setting `disable_automatic_requirements_detection=True`.
+
     Attributes:
         parent_image: Full name of the Docker image that should be
             used as the parent for the image that will be built. Defaults to
@@ -137,10 +146,29 @@ class DockerSettings(BaseSettings):
             packages.
         python_package_installer_args: Arguments to pass to the python package
             installer.
-        replicate_local_python_environment: If not `None`, ZenML will use the
-            specified method to generate a requirements file that replicates
-            the packages installed in the currently running python environment.
-            This requirements file will then be installed in the Docker image.
+        disable_automatic_requirements_detection: If set to True, ZenML will
+            not automatically detect requirements.txt files or pyproject.toml
+            files in your source root.
+        replicate_local_python_environment: If set to True, ZenML will run
+            `pip freeze` to gather the requirements of the local Python
+            environment and then install them in the Docker image.
+        pyproject_path: Path to a pyproject.toml file. If given, the
+            dependencies will be exported to a requirements.txt
+            formatted file using the `pyproject_export_command` and then
+            installed inside the Docker image.
+        pyproject_export_command: Command to export the dependencies inside a
+            pyproject.toml file to a requirements.txt formatted file. If not
+            given and ZenML needs to export the requirements anyway, `uv export`
+            and `poetry export` will be tried to see if one of them works. This
+            command can contain a `{directory}` placeholder which will be
+            replaced with the directory in which the pyproject.toml file is
+            stored.
+            **Note**: This command will be run before any code files are copied
+            into the image. It is therefore not possible to install a local
+            project using this command. This command should exclude any local
+            projects, and you can specify a `local_project_install_command`
+            instead which will be run after the code files are copied into the
+            image.
         requirements: Path to a requirements file or a list of required pip
             packages. During the image build, these requirements will be
             installed using pip. If you need to use a different tool to
@@ -149,13 +177,16 @@ class DockerSettings(BaseSettings):
         required_integrations: List of ZenML integrations that should be
             installed. All requirements for the specified integrations will
             be installed inside the Docker image.
-        required_hub_plugins: DEPRECATED/UNUSED.
         install_stack_requirements: If `True`, ZenML will automatically detect
             if components of your active stack are part of a ZenML integration
             and install the corresponding requirements and apt packages.
             If you set this to `False` or use custom components in your stack,
             you need to make sure these get installed by specifying them in
             the `requirements` and `apt_packages` attributes.
+        local_project_install_command: Command to install a local project in
+            the Docker image. This is run after the code files are copied into
+            the image, and it is therefore only possible when code is included
+            in the image, not downloaded at runtime.
         apt_packages: APT packages to install inside the Docker image.
         environment: Dictionary of environment variables to set inside the
             Docker image.
@@ -170,14 +201,6 @@ class DockerSettings(BaseSettings):
             from a code repository if possible.
         allow_download_from_artifact_store: If `True`, code can be downloaded
             from the artifact store.
-        build_options: DEPRECATED, use parent_image_build_config.build_options
-            instead.
-        dockerignore: DEPRECATED, use build_config.dockerignore instead.
-        copy_files: DEPRECATED/UNUSED.
-        copy_global_config: DEPRECATED/UNUSED.
-        source_files: DEPRECATED. Use allow_including_files_in_images,
-            allow_download_from_code_repository and
-            allow_download_from_artifact_store instead.
     """
 
     parent_image: Optional[str] = None
@@ -191,14 +214,18 @@ class DockerSettings(BaseSettings):
         PythonPackageInstaller.PIP
     )
     python_package_installer_args: Dict[str, Any] = {}
+    disable_automatic_requirements_detection: bool = True
     replicate_local_python_environment: Optional[
-        Union[List[str], PythonEnvironmentExportMethod]
+        Union[List[str], PythonEnvironmentExportMethod, bool]
     ] = Field(default=None, union_mode="left_to_right")
+    pyproject_path: Optional[str] = None
+    pyproject_export_command: Optional[List[str]] = None
     requirements: Union[None, str, List[str]] = Field(
         default=None, union_mode="left_to_right"
     )
     required_integrations: List[str] = []
     install_stack_requirements: bool = True
+    local_project_install_command: Optional[str] = None
     apt_packages: List[str] = []
     environment: Dict[str, Any] = {}
     user: Optional[str] = None
@@ -221,6 +248,8 @@ class DockerSettings(BaseSettings):
         "copy_global_config",
         "source_files",
         "required_hub_plugins",
+        "build_options",
+        "dockerignore",
     )
 
     @model_validator(mode="before")
@@ -305,6 +334,58 @@ class DockerSettings(BaseSettings):
 
         return self
 
+    @model_validator(mode="after")
+    def _validate_code_files_included_if_installing_local_project(
+        self,
+    ) -> "DockerSettings":
+        """Ensures that files are included when installing a local package.
+
+        Raises:
+            ValueError: If files are not included in the Docker image
+                when trying to install a local package.
+
+        Returns:
+            The validated settings values.
+        """
+        if (
+            self.local_project_install_command
+            and not self.allow_including_files_in_images
+        ):
+            raise ValueError(
+                "Files must be included in the Docker image when trying to "
+                "install a local python package. You can do so by setting "
+                "the `allow_including_files_in_images` attribute of your "
+                "DockerSettings to `True`."
+            )
+
+        return self
+
+    @model_validator(mode="after")
+    def _deprecate_replicate_local_environment_commands(
+        self,
+    ) -> "DockerSettings":
+        """Deprecates some values for `replicate_local_python_environment`.
+
+        Returns:
+            The validated settings values.
+        """
+        if isinstance(
+            self.replicate_local_python_environment,
+            (str, list, PythonEnvironmentExportMethod),
+        ):
+            logger.warning(
+                "Specifying a command (`%s`) for "
+                "`DockerSettings.replicate_local_python_environment` is "
+                "deprecated. If you want to replicate your exact local "
+                "environment using `pip freeze`, set "
+                "`DockerSettings.replicate_local_python_environment=True`. "
+                "If you want to export requirements from a pyproject.toml "
+                "file, use `DockerSettings.pyproject_path` and "
+                "`DockerSettings.pyproject_export_command` instead."
+            )
+
+        return self
+
     model_config = ConfigDict(
         # public attributes are immutable
         frozen=True,

zenml/constants.py

@@ -341,6 +341,7 @@ DEFAULT_ZENML_SERVER_REPORT_USER_ACTIVITY_TO_DB_SECONDS = 30
 DEFAULT_ZENML_SERVER_MAX_REQUEST_BODY_SIZE_IN_BYTES = 256 * 1024 * 1024
 
 DEFAULT_REPORTABLE_RESOURCES = ["project", "pipeline", "pipeline_run", "model"]
+RUN_TEMPLATE_TRIGGERS_FEATURE_NAME = "template_run"
 
 # API Endpoint paths:
 ACTIVATE = "/activate"

zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py

@@ -15,7 +15,7 @@
 
 from typing import TYPE_CHECKING, Optional, Type
 
-from pydantic import PositiveInt
+from pydantic import NonNegativeInt, PositiveInt
 
 from zenml.config.base_settings import BaseSettings
 from zenml.constants import KUBERNETES_CLUSTER_RESOURCE_TYPE
@@ -59,6 +59,14 @@ class KubernetesOrchestratorSettings(BaseSettings):
         pod_failure_backoff: The backoff factor for pod failure retries and
             pod startup retries.
         max_parallelism: Maximum number of steps to run in parallel.
+        successful_jobs_history_limit: The number of successful jobs
+            to retain. This only applies to jobs created when scheduling a
+            pipeline.
+        failed_jobs_history_limit: The number of failed jobs to retain.
+            This only applies to jobs created when scheduling a pipeline.
+        ttl_seconds_after_finished: The amount of seconds to keep finished jobs
+            before deleting them. This only applies to jobs created when
+            scheduling a pipeline.
     """
 
     synchronous: bool = True
@@ -74,6 +82,9 @@ class KubernetesOrchestratorSettings(BaseSettings):
     pod_failure_retry_delay: int = 10
     pod_failure_backoff: float = 1.0
     max_parallelism: Optional[PositiveInt] = None
+    successful_jobs_history_limit: Optional[NonNegativeInt] = None
+    failed_jobs_history_limit: Optional[NonNegativeInt] = None
+    ttl_seconds_after_finished: Optional[NonNegativeInt] = None
 
 
 class KubernetesOrchestratorConfig(

zenml/integrations/kubernetes/orchestrators/kube_utils.py

@@ -34,7 +34,7 @@ Adjusted from https://github.com/tensorflow/tfx/blob/master/tfx/utils/kube_utils
 import enum
 import re
 import time
-from typing import Any, Callable, Dict, Optional, TypeVar, cast
+from typing import Any, Callable, Dict, List, Optional, TypeVar, cast
 
 from kubernetes import client as k8s_client
 from kubernetes import config as k8s_config
@@ -554,3 +554,26 @@ def create_and_wait_for_pod_to_start(
         total_wait += delay
         time.sleep(delay)
         delay *= startup_failure_backoff
+
+
+def get_pod_owner_references(
+    core_api: k8s_client.CoreV1Api, pod_name: str, namespace: str
+) -> List[k8s_client.V1OwnerReference]:
+    """Get owner references for a pod.
+
+    Args:
+        core_api: Kubernetes CoreV1Api client.
+        pod_name: Name of the pod.
+        namespace: Kubernetes namespace.
+
+    Returns:
+        List of owner references.
+    """
+    pod = get_pod(core_api=core_api, pod_name=pod_name, namespace=namespace)
+
+    if not pod or not pod.metadata or not pod.metadata.owner_references:
+        return []
+
+    return cast(
+        List[k8s_client.V1OwnerReference], pod.metadata.owner_references
+    )

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py

@@ -523,6 +523,9 @@ class KubernetesOrchestrator(ContainerizedOrchestrator):
                 pod_settings=orchestrator_pod_settings,
                 env=environment,
                 mount_local_stores=self.config.is_local,
+                successful_jobs_history_limit=settings.successful_jobs_history_limit,
+                failed_jobs_history_limit=settings.failed_jobs_history_limit,
+                ttl_seconds_after_finished=settings.ttl_seconds_after_finished,
             )
 
             self._k8s_batch_api.create_namespaced_cron_job(

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py

@@ -71,7 +71,7 @@ def main() -> None:
     # Parse / extract args.
     args = parse_args()
 
-    orchestrator_run_id = socket.gethostname()
+    orchestrator_pod_name = socket.gethostname()
 
     client = Client()
 
@@ -95,7 +95,22 @@ def main() -> None:
     core_api = k8s_client.CoreV1Api(kube_client)
 
     env = get_config_environment_vars()
-    env[ENV_ZENML_KUBERNETES_RUN_ID] = orchestrator_run_id
+    env[ENV_ZENML_KUBERNETES_RUN_ID] = orchestrator_pod_name
+
+    try:
+        owner_references = kube_utils.get_pod_owner_references(
+            core_api=core_api,
+            pod_name=orchestrator_pod_name,
+            namespace=args.kubernetes_namespace,
+        )
+    except Exception as e:
+        logger.warning(f"Failed to get pod owner references: {str(e)}")
+        owner_references = []
+    else:
+        # Make sure None of the owner references are marked as controllers of
+        # the created pod, which messes with the garbage collection logic.
+        for owner_reference in owner_references:
+            owner_reference.controller = False
 
     def run_step_on_kubernetes(step_name: str) -> None:
         """Run a pipeline step in a separate Kubernetes pod.
@@ -112,7 +127,7 @@ def main() -> None:
             settings.model_dump() if settings else {}
         )
 
-        if settings.pod_name_prefix and not orchestrator_run_id.startswith(
+        if settings.pod_name_prefix and not orchestrator_pod_name.startswith(
             settings.pod_name_prefix
         ):
             max_length = (
@@ -125,7 +140,7 @@ def main() -> None:
             )
             pod_name = f"{pod_name_prefix}-{step_name}"
         else:
-            pod_name = f"{orchestrator_run_id}-{step_name}"
+            pod_name = f"{orchestrator_pod_name}-{step_name}"
 
         pod_name = kube_utils.sanitize_pod_name(
             pod_name, namespace=args.kubernetes_namespace
@@ -179,6 +194,7 @@ def main() -> None:
             service_account_name=settings.step_pod_service_account_name
             or settings.service_account_name,
             mount_local_stores=mount_local_stores,
+            owner_references=owner_references,
         )
 
         kube_utils.create_and_wait_for_pod_to_start(
@@ -231,7 +247,7 @@ def main() -> None:
             else:
                 # For a run triggered by a schedule, we can only use the
                 # orchestrator run ID to find the pipeline run.
-                list_args = dict(orchestrator_run_id=orchestrator_run_id)
+                list_args = dict(orchestrator_run_id=orchestrator_pod_name)
 
             pipeline_runs = client.list_pipeline_runs(
                 hydrate=True,

zenml/integrations/kubernetes/orchestrators/manifest_utils.py

@@ -107,6 +107,7 @@ def build_pod_manifest(
     service_account_name: Optional[str] = None,
     env: Optional[Dict[str, str]] = None,
     mount_local_stores: bool = False,
+    owner_references: Optional[List[k8s_client.V1OwnerReference]] = None,
 ) -> k8s_client.V1Pod:
     """Build a Kubernetes pod manifest for a ZenML run or step.
 
@@ -125,6 +126,7 @@ def build_pod_manifest(
         env: Environment variables to set.
         mount_local_stores: Whether to mount the local stores path inside the
             pod.
+        owner_references: List of owner references for the pod.
 
     Returns:
         Pod manifest.
@@ -180,6 +182,7 @@ def build_pod_manifest(
     pod_metadata = k8s_client.V1ObjectMeta(
         name=pod_name,
         labels=labels,
+        owner_references=owner_references,
     )
 
     if pod_settings and pod_settings.annotations:
@@ -279,6 +282,9 @@ def build_cron_job_manifest(
     service_account_name: Optional[str] = None,
     env: Optional[Dict[str, str]] = None,
     mount_local_stores: bool = False,
+    successful_jobs_history_limit: Optional[int] = None,
+    failed_jobs_history_limit: Optional[int] = None,
+    ttl_seconds_after_finished: Optional[int] = None,
 ) -> k8s_client.V1CronJob:
     """Create a manifest for launching a pod as scheduled CRON job.
 
@@ -298,6 +304,10 @@ def build_cron_job_manifest(
         env: Environment variables to set.
         mount_local_stores: Whether to mount the local stores path inside the
             pod.
+        successful_jobs_history_limit: The number of successful jobs to retain.
+        failed_jobs_history_limit: The number of failed jobs to retain.
+        ttl_seconds_after_finished: The amount of seconds to keep finished jobs
+            before deleting them.
 
     Returns:
         CRON job manifest.
@@ -318,6 +328,8 @@ def build_cron_job_manifest(
 
     job_spec = k8s_client.V1CronJobSpec(
         schedule=cron_expression,
+        successful_jobs_history_limit=successful_jobs_history_limit,
+        failed_jobs_history_limit=failed_jobs_history_limit,
         job_template=k8s_client.V1JobTemplateSpec(
             metadata=pod_manifest.metadata,
             spec=k8s_client.V1JobSpec(
@@ -325,6 +337,7 @@ def build_cron_job_manifest(
                     metadata=pod_manifest.metadata,
                     spec=pod_manifest.spec,
                 ),
+                ttl_seconds_after_finished=ttl_seconds_after_finished,
             ),
         ),
     )

zenml/orchestrators/step_launcher.py

@@ -292,6 +292,10 @@ class StepLauncher:
                             artifacts=step_run.outputs,
                             model_version=model_version,
                         )
+                    step_run_utils.cascade_tags_for_output_artifacts(
+                        artifacts=step_run.outputs,
+                        tags=pipeline_run.config.tags,
+                    )
 
         except:  # noqa: E722
             logger.error(f"Pipeline run `{pipeline_run.name}` failed.")

zenml/orchestrators/step_run_utils.py

@@ -13,8 +13,9 @@
 #  permissions and limitations under the License.
 """Utilities for creating step runs."""
 
-from typing import Dict, List, Optional, Set, Tuple
+from typing import Dict, List, Optional, Set, Tuple, Union
 
+from zenml import Tag, add_tags
 from zenml.client import Client
 from zenml.config.step_configurations import Step
 from zenml.constants import CODE_HASH_PARAMETER_NAME, TEXT_FIELD_MAX_LENGTH
@@ -333,6 +334,11 @@ def create_cached_step_runs(
                     model_version=model_version,
                 )
 
+            cascade_tags_for_output_artifacts(
+                artifacts=step_run.outputs,
+                tags=pipeline_run.config.tags,
+            )
+
             logger.info("Using cached version of step `%s`.", invocation_id)
             cached_invocations.add(invocation_id)
 
@@ -382,3 +388,26 @@ def link_output_artifacts_to_model_version(
                 artifact_version=output_artifact,
                 model_version=model_version,
             )
+
+
+def cascade_tags_for_output_artifacts(
+    artifacts: Dict[str, List[ArtifactVersionResponse]],
+    tags: Optional[List[Union[str, Tag]]] = None,
+) -> None:
+    """Tag the outputs of a step run.
+
+    Args:
+        artifacts: The step output artifacts.
+        tags: The tags to add to the artifacts.
+    """
+    if tags is None:
+        return
+
+    cascade_tags = [t for t in tags if isinstance(t, Tag) and t.cascade]
+
+    for output_artifacts in artifacts.values():
+        for output_artifact in output_artifacts:
+            add_tags(
+                tags=[t.name for t in cascade_tags],
+                artifact_version_id=output_artifact.id,
+            )

zenml/pipelines/build_utils.py

@@ -80,6 +80,11 @@ def requires_included_code(
     for step in deployment.step_configurations.values():
         docker_settings = step.config.docker_settings
 
+        if docker_settings.local_project_install_command:
+            # When installing a local package, we need to include the code
+            # files in the container image.
+            return True
+
         if docker_settings.allow_download_from_artifact_store:
             return False
 
@@ -136,6 +141,9 @@ def code_download_possible(
         Whether code download is possible for the deployment.
     """
     for step in deployment.step_configurations.values():
+        if step.config.docker_settings.local_project_install_command:
+            return False
+
         if step.config.docker_settings.allow_download_from_artifact_store:
             continue
 

zenml/utils/pipeline_docker_image_builder.py

@@ -68,6 +68,11 @@ PIP_DEFAULT_ARGS = {
 }
 UV_DEFAULT_ARGS = {"no-cache-dir": None}
 
+DEFAULT_PYPROJECT_EXPORT_COMMANDS = [
+    "uv export --format=requirements-txt --directory={directory} --no-hashes --no-emit-project",
+    "poetry export --format=requirements.txt --directory={directory} --without-hashes",
+]
+
 
 class PipelineDockerImageBuilder:
     """Builds Docker images to run a ZenML pipeline."""
@@ -156,10 +161,11 @@ class PipelineDockerImageBuilder:
         requires_zenml_build = any(
             [
                 docker_settings.requirements,
+                docker_settings.pyproject_path,
                 docker_settings.required_integrations,
-                docker_settings.required_hub_plugins,
                 docker_settings.replicate_local_python_environment,
                 docker_settings.install_stack_requirements,
+                docker_settings.local_project_install_command,
                 docker_settings.apt_packages,
                 docker_settings.environment,
                 include_files,
@@ -434,8 +440,56 @@ class PipelineDockerImageBuilder:
             - Packages installed in the local Python environment
             - Requirements defined by stack integrations
             - Requirements defined by user integrations
+            - Requirements exported from a pyproject.toml
             - User-defined requirements
         """
+        implicit_requirements = False
+        pyproject_path = docker_settings.pyproject_path
+        requirements = docker_settings.requirements
+
+        if not any(
+            [
+                docker_settings.replicate_local_python_environment,
+                requirements,
+                pyproject_path,
+            ]
+        ):
+            root = source_utils.get_source_root()
+            requirements_path = os.path.join(root, "requirements.txt")
+            pyproject_file_path = os.path.join(root, "pyproject.toml")
+
+            if os.path.exists(requirements_path):
+                implicit_requirements = True
+                requirements = requirements_path
+            elif os.path.exists(pyproject_file_path):
+                implicit_requirements = True
+                pyproject_path = pyproject_file_path
+
+        if (
+            implicit_requirements
+            and docker_settings.disable_automatic_requirements_detection
+        ):
+            # TODO: This is only temporary to log a warning notifying users
+            # that we will soon switch the default behavior to
+            # `disable_automatic_requirements_detection=False`. Remove and
+            # adjust the logic once we've made this change.
+            if log:
+                logger.warning(
+                    "Detected `requirements.txt` or `pyproject.toml` files in "
+                    "the source root. In future versions of ZenML, these will "
+                    "be automatically picked up and installed in Docker images "
+                    "by default. To disable this behavior and keep the "
+                    "current behavior, set "
+                    "`DockerSettings.disable_automatic_requirements_detection` "
+                    "to `True`. If you want to enable this behavior right away, "
+                    "you can do so by setting "
+                    "`DockerSettings.disable_automatic_requirements_detection` "
+                    "to `False`."
+                )
+            implicit_requirements = False
+            requirements = None
+            pyproject_path = None
+
         requirements_files: List[Tuple[str, str, List[str]]] = []
 
         # Generate requirements file for the local environment if configured
@@ -447,6 +501,10 @@ class PipelineDockerImageBuilder:
                 command = (
                     docker_settings.replicate_local_python_environment.command
                 )
+            elif isinstance(
+                docker_settings.replicate_local_python_environment, bool
+            ):
+                command = PythonEnvironmentExportMethod.PIP_FREEZE.command
             else:
                 command = " ".join(
                     docker_settings.replicate_local_python_environment
@@ -520,9 +578,69 @@ class PipelineDockerImageBuilder:
                     ", ".join(f"`{r}`" for r in integration_requirements_list),
                 )
 
+        if pyproject_path:
+            path = os.path.abspath(pyproject_path)
+
+            if not os.path.exists(path):
+                raise FileNotFoundError(
+                    f"Pyproject file {path} does not exist."
+                )
+
+            def _run_command(command: str) -> str:
+                command = command.format(directory=os.path.dirname(path))
+
+                result = subprocess.run(
+                    command,
+                    capture_output=True,
+                    check=True,
+                    shell=True,  # nosec
+                )
+                return result.stdout.decode()
+
+            if docker_settings.pyproject_export_command:
+                command = " ".join(docker_settings.pyproject_export_command)
+
+                try:
+                    pyproject_requirements = _run_command(command)
+                except subprocess.CalledProcessError as e:
+                    raise RuntimeError(
+                        "Failed to export pyproject dependencies with "
+                        f"command `{command}`: {e.stderr.decode()}"
+                    )
+            else:
+                command_errors = {}
+
+                for command in DEFAULT_PYPROJECT_EXPORT_COMMANDS:
+                    try:
+                        pyproject_requirements = _run_command(command)
+                    except subprocess.CalledProcessError as e:
+                        command_errors[command] = e.stderr.decode()
+                    else:
+                        break
+                else:
+                    raise RuntimeError(
+                        "Failed to export pyproject dependencies with the "
+                        f"following commands: {command_errors}. Please specify "
+                        "a working command to export your pyproject.toml "
+                        "dependencies to a requirements.txt formatted file "
+                        "using `DockerSettings.pyproject_export_command`."
+                    )
+
+            requirements_files.append(
+                (".zenml_pyproject_requirements", pyproject_requirements, [])
+            )
+            if log:
+                logger.info(
+                    "- %s python packages from file `%s`",
+                    "Implicitly including"
+                    if implicit_requirements
+                    else "Including",
+                    path,
+                )
+
         # Generate/Read requirements file for user-defined requirements
-        if isinstance(docker_settings.requirements, str):
-            path = os.path.abspath(docker_settings.requirements)
+        if isinstance(requirements, str):
+            path = os.path.abspath(requirements)
             try:
                 user_requirements = io_utils.read_file_contents_as_string(path)
             except FileNotFoundError as e:
@@ -531,15 +649,18 @@ class PipelineDockerImageBuilder:
                 ) from e
             if log:
                 logger.info(
-                    "- Including user-defined requirements from file `%s`",
+                    "- %s user-defined requirements from file `%s`",
+                    "Implicitly including"
+                    if implicit_requirements
+                    else "Including",
                     path,
                 )
-        elif isinstance(docker_settings.requirements, List):
-            user_requirements = "\n".join(docker_settings.requirements)
+        elif isinstance(requirements, List):
+            user_requirements = "\n".join(requirements)
             if log:
                 logger.info(
                     "- Including user-defined requirements: %s",
-                    ", ".join(f"`{r}`" for r in docker_settings.requirements),
+                    ", ".join(f"`{r}`" for r in requirements),
                 )
         else:
             user_requirements = None
@@ -638,6 +759,11 @@ class PipelineDockerImageBuilder:
         lines.append("COPY . .")
         lines.append("RUN chmod -R a+rw .")
 
+        if docker_settings.local_project_install_command:
+            lines.append(
+                f"RUN {docker_settings.local_project_install_command}"
+            )
+
         if docker_settings.user:
             # Change file ownership to specified user
             lines.append(f"RUN chown -R {docker_settings.user} .")

zenml/utils/tag_utils.py

@@ -348,10 +348,10 @@ def add_tags(
             if isinstance(tag, Tag):
                 tag_model = client.get_tag(tag.name)
 
-                if tag.exclusive != tag_model.exclusive:
+                if bool(tag.exclusive) != tag_model.exclusive:
                     raise ValueError(
-                        f"The tag `{tag.name}` is an "
-                        f"{'exclusive' if tag_model.exclusive else 'non-exclusive'} "
+                        f"The tag `{tag.name}` is "
+                        f"{'an exclusive' if tag_model.exclusive else 'a non-exclusive'} "
                         "tag. Please update it before attaching it to a resource."
                     )
                 if tag.cascade is not None:

zenml/zen_server/feature_gate/endpoint_utils.py

@@ -15,47 +15,44 @@
 
 from uuid import UUID
 
-from zenml.zen_server.rbac.models import ResourceType
 from zenml.zen_server.utils import feature_gate, server_config
 
 
-def check_entitlement(resource_type: ResourceType) -> None:
+def check_entitlement(feature: str) -> None:
     """Queries the feature gate to see if the operation falls within the Pro workspaces entitlements.
 
     Raises an exception if the user is not entitled to create an instance of the
     resource. Otherwise, simply returns.
 
     Args:
-        resource_type: The type of resource to check for.
+        feature: The feature to check for.
     """
     if not server_config().feature_gate_enabled:
         return
-    return feature_gate().check_entitlement(resource=resource_type)
+    return feature_gate().check_entitlement(feature=feature)
 
 
-def report_usage(resource_type: ResourceType, resource_id: UUID) -> None:
+def report_usage(feature: str, resource_id: UUID) -> None:
     """Reports the creation/usage of a feature/resource.
 
     Args:
-        resource_type: The type of resource to report a usage for
+        feature: The feature to report a usage for.
         resource_id: ID of the resource that was created.
     """
     if not server_config().feature_gate_enabled:
         return
-    feature_gate().report_event(
-        resource=resource_type, resource_id=resource_id
-    )
+    feature_gate().report_event(feature=feature, resource_id=resource_id)
 
 
-def report_decrement(resource_type: ResourceType, resource_id: UUID) -> None:
+def report_decrement(feature: str, resource_id: UUID) -> None:
     """Reports the deletion/deactivation of a feature/resource.
 
     Args:
-        resource_type: The type of resource to report a decrement in count for.
+        feature: The feature to report a decrement in count for.
         resource_id: ID of the resource that was deleted.
     """
     if not server_config().feature_gate_enabled:
         return
     feature_gate().report_event(
-        resource=resource_type, resource_id=resource_id, is_decrement=True
+        feature=feature, resource_id=resource_id, is_decrement=True
     )

zenml/zen_server/feature_gate/feature_gate_interface.py

@@ -16,18 +16,16 @@
 from abc import ABC, abstractmethod
 from uuid import UUID
 
-from zenml.zen_server.rbac.models import ResourceType
-
 
 class FeatureGateInterface(ABC):
     """Feature gate interface definition."""
 
     @abstractmethod
-    def check_entitlement(self, resource: ResourceType) -> None:
+    def check_entitlement(self, feature: str) -> None:
         """Checks if a user is entitled to create a resource.
 
         Args:
-            resource: The resource the user wants to create
+            feature: The feature the user wants to use.
 
         Raises:
             UpgradeRequiredError in case a subscription limit is reached
@@ -36,14 +34,14 @@ class FeatureGateInterface(ABC):
     @abstractmethod
     def report_event(
         self,
-        resource: ResourceType,
+        feature: str,
         resource_id: UUID,
         is_decrement: bool = False,
     ) -> None:
         """Reports the usage of a feature to the aggregator backend.
 
         Args:
-            resource: The resource the user created
+            feature: The feature the user used.
             resource_id: ID of the resource that was created/deleted.
             is_decrement: In case this event reports an actual decrement of usage
         """

zenml/zen_server/feature_gate/zenml_cloud_feature_gate.py

@@ -25,7 +25,6 @@ from zenml.zen_server.cloud_utils import cloud_connection
 from zenml.zen_server.feature_gate.feature_gate_interface import (
     FeatureGateInterface,
 )
-from zenml.zen_server.rbac.models import ResourceType
 
 logger = get_logger(__name__)
 
@@ -47,7 +46,7 @@ class RawUsageEvent(BaseModel):
     organization_id: str = Field(
         description="The organization that this usage can be attributed to.",
     )
-    feature: ResourceType = Field(
+    feature: str = Field(
         description="The feature whose usage is being reported.",
     )
     total: int = Field(
@@ -66,22 +65,22 @@ class ZenMLCloudFeatureGateInterface(FeatureGateInterface):
         """Initialize the object."""
         self._connection = cloud_connection()
 
-    def check_entitlement(self, resource: ResourceType) -> None:
+    def check_entitlement(self, feature: str) -> None:
         """Checks if a user is entitled to create a resource.
 
         Args:
-            resource: The resource the user wants to create
+            feature: The feature the user wants to use.
 
         Raises:
             SubscriptionUpgradeRequiredError: in case a subscription limit is reached
         """
         try:
             response = self._connection.get(
-                endpoint=ENTITLEMENT_ENDPOINT + "/" + resource, params=None
+                endpoint=ENTITLEMENT_ENDPOINT + "/" + feature, params=None
             )
         except SubscriptionUpgradeRequiredError:
             raise SubscriptionUpgradeRequiredError(
-                f"Your subscription reached its `{resource}` limit. Please "
+                f"Your subscription reached its `{feature}` limit. Please "
                 f"upgrade your subscription or reach out to us."
             )
 
@@ -94,20 +93,20 @@ class ZenMLCloudFeatureGateInterface(FeatureGateInterface):
 
     def report_event(
         self,
-        resource: ResourceType,
+        feature: str,
         resource_id: UUID,
         is_decrement: bool = False,
     ) -> None:
         """Reports the usage of a feature to the aggregator backend.
 
         Args:
-            resource: The resource the user created
+            feature: The feature the user used.
             resource_id: ID of the resource that was created/deleted.
             is_decrement: In case this event reports an actual decrement of usage
         """
         data = RawUsageEvent(
             organization_id=ORGANIZATION_ID,
-            feature=resource,
+            feature=feature,
             total=1 if not is_decrement else -1,
             metadata={
                 "workspace_id": str(server_config.get_external_server_id()),

zenml/zen_server/rbac/endpoint_utils.py

@@ -182,7 +182,7 @@ def verify_permissions_and_get_or_create_entity(
     def _pre_creation_hook() -> None:
         verify_permission_for_model(model=request_model, action=Action.CREATE)
         if resource_type and needs_usage_increment:
-            check_entitlement(resource_type=resource_type)
+            check_entitlement(feature=resource_type)
 
     model, created = get_or_create_method(request_model, _pre_creation_hook)
 

zenml/zen_server/rbac/rbac_sql_zen_store.py

@@ -67,7 +67,7 @@ class RBACSqlZenStore(SqlZenStore):
                 action=Action.CREATE,
                 project_id=model_request.project,
             )
-            check_entitlement(resource_type=ResourceType.MODEL)
+            check_entitlement(feature=ResourceType.MODEL)
         except Exception as e:
             allow_model_creation = False
             error = e
@@ -92,7 +92,7 @@ class RBACSqlZenStore(SqlZenStore):
 
         if created:
             report_usage(
-                resource_type=ResourceType.MODEL, resource_id=model_response.id
+                feature=ResourceType.MODEL, resource_id=model_response.id
             )
         else:
             verify_permission_for_model(model_response, action=Action.READ)

zenml/zen_server/routers/run_templates_endpoints.py

@@ -25,7 +25,12 @@ from fastapi import (
 from zenml.analytics.enums import AnalyticsEvent
 from zenml.analytics.utils import track_handler
 from zenml.config.pipeline_run_configuration import PipelineRunConfiguration
-from zenml.constants import API, RUN_TEMPLATES, VERSION_1
+from zenml.constants import (
+    API,
+    RUN_TEMPLATE_TRIGGERS_FEATURE_NAME,
+    RUN_TEMPLATES,
+    VERSION_1,
+)
 from zenml.models import (
     Page,
     PipelineRunResponse,
@@ -36,6 +41,9 @@ from zenml.models import (
 )
 from zenml.zen_server.auth import AuthContext, authorize
 from zenml.zen_server.exceptions import error_response
+from zenml.zen_server.feature_gate.endpoint_utils import (
+    check_entitlement,
+)
 from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_create_entity,
     verify_permissions_and_delete_entity,
@@ -269,6 +277,7 @@ if server_config().workload_manager_enabled:
                 action=Action.CREATE,
                 project_id=template.project.id,
             )
+            check_entitlement(feature=RUN_TEMPLATE_TRIGGERS_FEATURE_NAME)
 
             return run_template(
                 template=template,

zenml/zen_server/template_execution/utils.py

@@ -24,6 +24,7 @@ from zenml.constants import (
     ENV_ZENML_RUNNER_IMAGE_DISABLE_UV,
     ENV_ZENML_RUNNER_PARENT_IMAGE,
     ENV_ZENML_RUNNER_POD_TIMEOUT,
+    RUN_TEMPLATE_TRIGGERS_FEATURE_NAME,
     handle_bool_env_var,
     handle_int_env_var,
 )
@@ -49,7 +50,11 @@ from zenml.pipelines.run_utils import (
 )
 from zenml.stack.flavor import Flavor
 from zenml.utils import pydantic_utils, requirements_utils, settings_utils
+from zenml.utils.time_utils import utc_now
 from zenml.zen_server.auth import AuthContext, generate_access_token
+from zenml.zen_server.feature_gate.endpoint_utils import (
+    report_usage,
+)
 from zenml.zen_server.template_execution.runner_entrypoint_configuration import (
     RunnerEntrypointConfiguration,
 )
@@ -190,6 +195,11 @@ def run_template(
     placeholder_run = create_placeholder_run(deployment=new_deployment)
     assert placeholder_run
 
+    report_usage(
+        feature=RUN_TEMPLATE_TRIGGERS_FEATURE_NAME,
+        resource_id=placeholder_run.id,
+    )
+
     # We create an API token scoped to the pipeline run that never expires
     api_token = generate_access_token(
         user_id=auth_context.user.id,
@@ -300,7 +310,8 @@ def run_template(
                 zen_store().update_run(
                     run_id=placeholder_run.id,
                     run_update=PipelineRunUpdate(
-                        status=ExecutionStatus.FAILED
+                        status=ExecutionStatus.FAILED,
+                        end_time=utc_now(),
                     ),
                 )
                 raise

zenml/zen_stores/sql_zen_store.py

@@ -11354,13 +11354,10 @@ class SqlZenStore(BaseZenStore):
             except EntityExistsError:
                 if isinstance(tag, tag_utils.Tag):
                     tag_schema = self._get_tag_schema(tag.name, session)
-                    if (
-                        tag.exclusive is not None
-                        and tag.exclusive != tag_schema.exclusive
-                    ):
+                    if bool(tag.exclusive) != tag_schema.exclusive:
                         raise ValueError(
-                            f"Tag `{tag_schema.name}` has been defined as a "
-                            f"{'exclusive' if tag_schema.exclusive else 'non-exclusive'} "
+                            f"Tag `{tag_schema.name}` has been defined as "
+                            f"{'an exclusive' if tag_schema.exclusive else 'a non-exclusive'} "
                             "tag. Please update it before attaching it to resources."
                         )
                 else:

{zenml_nightly-0.82.0.dev20250513.dist-info → zenml_nightly-0.82.0.dev20250514.dist-info}/METADATA

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

{zenml_nightly-0.82.0.dev20250513.dist-info → zenml_nightly-0.82.0.dev20250514.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=GgvcZ4qW8lpLu66ULVW2IWU5xLriT6P_N-LWtNfqCL8,19
+zenml/VERSION,sha256=tqY-3ZZVi7BBuFj-AGEU8O9NpFcVRZl7FWjOi6F9yb8,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,10 +64,10 @@ 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=tEAIXwAGS8Flxo4mK5BGHsBk7LPWW16MOa6oRDymZ7k,6172
+zenml/config/build_configuration.py,sha256=me2jXT91Ndbrq6OFFTz2zbKimUWp6APQiE_goYONLgM,6417
 zenml/config/compiler.py,sha256=bK3LCDkrFc9SapJYH-vuQZ_o8scHNs-FdC5DblIUU4U,23024
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
-zenml/config/docker_settings.py,sha256=w62SDW2Qo5jFxcikHr8p-Lk6ymC2ieh3MnEfkte9Lf0,13322
+zenml/config/docker_settings.py,sha256=B3pYlnyQDiwSEHO6y965Jx6BMrVqKARjp7v23uBziLc,17068
 zenml/config/global_config.py,sha256=ZD3WodfcWMBJZOl1FNn3ztzwilGDDv9EKdHClLqSO8s,29562
 zenml/config/pipeline_configurations.py,sha256=7trCbElpqGGgawii2FrdLW8fKaAWCR8jACkNqdG_vcQ,3983
 zenml/config/pipeline_run_configuration.py,sha256=Y9C5mVUH-w4gc1w1PCQFpjEmfBBpSMvb8riA_sL78hY,2311
@@ -85,7 +85,7 @@ zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I
 zenml/config/store_config.py,sha256=Cla5p5dTB6nNlo8_OZDs9hod5hspi64vxwtZj882XgU,3559
 zenml/config/strict_base_model.py,sha256=t_ULrtJF2eW7TgyYBRobl1fscwwIZXATYky8ER97ev4,860
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
-zenml/constants.py,sha256=Lkm9EWkhAEwxdOuVHxSmLB6XEpa6-sw1TfKvpaVNEhk,16545
+zenml/constants.py,sha256=nLjJUhBqFHWHlGKbZZw-_tn7x99y6q4vmk5LOe7GhJI,16597
 zenml/container_registries/__init__.py,sha256=ZSPbBIOnzhg88kQSpYgKe_POLuru14m629665-kAVAA,2200
 zenml/container_registries/azure_container_registry.py,sha256=t1sfDa94Vzbyqtb1iPFNutJ2EXV5_p9CUNITasoiQ70,2667
 zenml/container_registries/base_container_registry.py,sha256=-9RIkD6oXNPaU59R3PB_PtyCqsFoLPLSn5xYZmEmzbc,8915
@@ -334,14 +334,14 @@ zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=ihM8s0
 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=bO_2RZ6TIO0Qp2zduha324m2zrg6cGw-ymK51Nm2E9o,9505
+zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=JH_Kxfh1eoiSlz-OpixJ6itEsDiGPvbkA5_PffMTa54,10215
 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=tzGhxFyf2of7soyEiEYAixuU9uDjT49-StAqhRJLMVo,25730
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=krsIXrHUk2EZAVpqMCn8KgICNhD_YtVlySarQD5C4bI,12039
+zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=8BVOpg40dGeRSkbzG_66xOksry5xpMVVmcXVDqb1bSs,18747
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=qubV13fSGE_vo8wSzIfcp66McclLw-zDtmB4yCCaS08,25974
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=Nd1YZsP6FZ4LTpoeFIM3YlFUugHX7F9axb7U9ZgGNdQ,12692
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=KjHfQK9VQEQkkkM2i9w51AzqolgIU01M5dgb2YGamvY,2754
-zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=Mte6q_NX0bU_JTivtN7ByQtfQX32iI2z2sElIpQhj7g,13271
+zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=QmK0HrAMalalcQMyG0rqBysoB3srQ4ywnKARdf-LtFc,14083
 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
@@ -690,14 +690,14 @@ zenml/orchestrators/local_docker/__init__.py,sha256=k8J68ydy6HmmvE9tWo32g761H8P_
 zenml/orchestrators/local_docker/local_docker_orchestrator.py,sha256=4b4oiCQ_PnfzgsHZuGrWeVEvjsrq_0yvVqtBUqz3Ai4,9353
 zenml/orchestrators/output_utils.py,sha256=01vqke1ZfmfuLpgxNerF-QL2wA0VPv1zUdvlMw0OwUY,3508
 zenml/orchestrators/publish_utils.py,sha256=_77t2sVq3Tm-yAYg2NPxcCD7Y8lwe0lm55NNrtMUzQQ,4741
-zenml/orchestrators/step_launcher.py,sha256=YOeQtvmI__c8ugrqk2KSyITYVOxoLIAZxUIxkLyv-UY,17857
-zenml/orchestrators/step_run_utils.py,sha256=7GOWAW3izFopGPkvvjYyJKiYgYuHuHJ8V6eZfwjAx8E,13401
+zenml/orchestrators/step_launcher.py,sha256=UM0TeFx_VtOy3Rq7uHRjljtABAezwiXYkUgGe6GGToA,18056
+zenml/orchestrators/step_run_utils.py,sha256=bG3ybu1hPt0IGe-7W4-4Qktb8EO7D7L1ChVMZAqEPmE,14256
 zenml/orchestrators/step_runner.py,sha256=EUgKG_g0fOQ6gnB1hPSSa6UXwUKVkguC-Yj-Q0yEQXg,26632
 zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7cM,5784
 zenml/orchestrators/utils.py,sha256=agh8m5v1OYU-16Rcp6YDwDbHt9S1F46zsRe3AWC_tBc,12967
 zenml/orchestrators/wheeled_orchestrator.py,sha256=eOnMcnd3sCzfhA2l6qRAzF0rOXzaojbjvvYvTkqixQo,4791
 zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,873
-zenml/pipelines/build_utils.py,sha256=FYflpfK8i9jLgy8Ybr2dhQZarUDA1_0shvEARPkBMYA,27591
+zenml/pipelines/build_utils.py,sha256=DltGesybT8qYum4i23mvWZlVRgp7UxWdbHd1Y9ySv5c,27889
 zenml/pipelines/pipeline_context.py,sha256=4BixReLcPo33VtNBDrMwnJqjKTinHjmO5AOfmoeIOQM,3659
 zenml/pipelines/pipeline_decorator.py,sha256=LB21QYrbFeBdUGwKBUNbdpXAxO4OOtYl5Vs_mzJNXqU,4600
 zenml/pipelines/pipeline_definition.py,sha256=954y2DmoY9luHhutC2nVRXzZ_KLnTWFnS5LSzrq_OZo,59637
@@ -784,7 +784,7 @@ zenml/utils/networking_utils.py,sha256=zTDbOMkMPRWiWLZ2ccchd37rvvZWIIh0Kr9dZE-VJ
 zenml/utils/notebook_utils.py,sha256=VBMU9Qnx_NdpB1TQtnej56L0hRr11fwniOva3ltUT90,4113
 zenml/utils/package_utils.py,sha256=wy0Mh8hHhOX2z1JfGN5lifG9yEsBQGLwNfur0M3J2tQ,2730
 zenml/utils/pagination_utils.py,sha256=TufckOqOKeDPwE3ySefL05zOzGUUA2Fqx_QFVhE2so0,1445
-zenml/utils/pipeline_docker_image_builder.py,sha256=QOZV_AYFbUtcfJZGNO7pH2_EoXyRqs9GZF_hTeoqW5E,25036
+zenml/utils/pipeline_docker_image_builder.py,sha256=Gp_eKg_LEdyBUSuZDC_PReyPHlqonKNwjxPJJLoAFY0,30256
 zenml/utils/proxy_utils.py,sha256=fgRlLa9pfXJDoxtB31_YP7DClOMQLek_nMmM0et6i3w,7241
 zenml/utils/pydantic_utils.py,sha256=oQcxY4VXuVY3n632atlvdmi12EYcSQ1xZuQJY3Je-sA,16592
 zenml/utils/requirements_utils.py,sha256=JqVChflTQwBMByRtKfsS3dV91QeWbdGYefcEBkTKXGk,2605
@@ -795,7 +795,7 @@ zenml/utils/singleton.py,sha256=uFRrUlUdS5VyY9lLJyl_n5kqppsqJLKkBhSj4g5VPkY,2757
 zenml/utils/source_code_utils.py,sha256=8iyNA2MGIORYVEkSdxNTXfS1ZdFKXTAG1dZRkeQtPL0,3751
 zenml/utils/source_utils.py,sha256=joKLghhDq9dh0fd8B0WRGX-nN-uwnGQdgmsyY_n-8gY,27033
 zenml/utils/string_utils.py,sha256=xJ8Abm52yFQyOpNrgpoLjDbPCgb6rpJsi8N4-7bb5GU,7254
-zenml/utils/tag_utils.py,sha256=C9VeVFB-WuhHhGG_iVsNghJrGjX8EKVuRJZs8hMhwpA,19172
+zenml/utils/tag_utils.py,sha256=2LJ-3XkXuWnffUsyfCwahzL_N2k4lIYIgBs1HVShhA0,19180
 zenml/utils/time_utils.py,sha256=-9Y9zwJ-6Gv7hoZQCoftPyC2LCLo2bYj6OgdyBaE44o,4076
 zenml/utils/typed_model.py,sha256=00EAo1I1VnOBHG4-ce8dPkyHRPpgi67SRIU-KdewRWs,4757
 zenml/utils/typing_utils.py,sha256=jP7JKrlLsuMIStXhwKFNWykE6SMOR72tJIJ_qEbQSNc,6555
@@ -1014,16 +1014,16 @@ zenml/zen_server/deploy/exceptions.py,sha256=tX0GNnLB_GMkeN7zGNlJRwtlrpZ5Slvyj_u
 zenml/zen_server/download_utils.py,sha256=g366TqytIWR4L4u9dd0jg-ouSWpDwdAh_-F2zryFzfo,4224
 zenml/zen_server/exceptions.py,sha256=hP8zJBofQljxPByFIa4tKpVjKdCAMJlKvk_5Opgusqg,9718
 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
-zenml/zen_server/feature_gate/zenml_cloud_feature_gate.py,sha256=aZHYgtJk-IDgucsawSWZsNrqBk2KQPmcn9VsPcacggg,4409
+zenml/zen_server/feature_gate/endpoint_utils.py,sha256=1LbOhY0D0KoJHnkWtSaN8DqUq_CWc8txLGa9029lvyw,2049
+zenml/zen_server/feature_gate/feature_gate_interface.py,sha256=4UdXaK-rVxpI9PKpp2GGWLV1-8TIhycAOaWooLwjcAo,1564
+zenml/zen_server/feature_gate/zenml_cloud_feature_gate.py,sha256=kek99YFYEhC927XBrqTGskri3tHDte05ifp00PH-Ef4,4315
 zenml/zen_server/jwt.py,sha256=N45kKuGLZWKR7j1CyPVpu6Xvmk1BUwZKiaIxJUl9rs4,6857
 zenml/zen_server/rate_limit.py,sha256=rOg5H_6rOGQ_qiSCtMKPREmL1LL3bZyn4czKILtImJg,6057
 zenml/zen_server/rbac/__init__.py,sha256=nACbn_G7nZt_AWM3zeFL0FCmELvQnvaOFMwvTG3-6ZE,637
-zenml/zen_server/rbac/endpoint_utils.py,sha256=l6V6DwsPGmT6dLJmY4MHMRfRXdc0lZVibD-Aea5UAHc,11604
+zenml/zen_server/rbac/endpoint_utils.py,sha256=cpukjGh4CA9flzuFKPI_dhQQl8nL8WVKXhcCTHrEOOE,11598
 zenml/zen_server/rbac/models.py,sha256=Xhz0XqmVqg3ji9uTwjlhQ4IuQ2ivrT4gVdHi5ZkjFC4,4933
 zenml/zen_server/rbac/rbac_interface.py,sha256=VPMNnUIPcqJWDngITX6cfaPlM6zJwzCApXAnT_oaxSQ,3431
-zenml/zen_server/rbac/rbac_sql_zen_store.py,sha256=djGycBkciChk48q8oL7mHM0c-cHJM-S7inv5f3gcDgg,5942
+zenml/zen_server/rbac/rbac_sql_zen_store.py,sha256=ZBjAbCUlhMsp8GZiIkULl2ztJ8K2Wf-0QdSxzQXwovM,5930
 zenml/zen_server/rbac/utils.py,sha256=09r1dSZ04hcag21rg6Ug2fMCdYfNjSBnx_W8C336GzE,24254
 zenml/zen_server/rbac/zenml_cloud_rbac.py,sha256=L1Tio4IcNNxNWnVpAqBWYgdVfrqR20ddvdhWAo1i3f4,6055
 zenml/zen_server/routers/__init__.py,sha256=ViyAhWL-ogHxE9wBvB_iMcur5H1NRVrzXkpogVY7FBA,641
@@ -1044,7 +1044,7 @@ zenml/zen_server/routers/pipelines_endpoints.py,sha256=R6OM-0Tc6ZNPzQmLs5MWTJIE4
 zenml/zen_server/routers/plugin_endpoints.py,sha256=vgIdA0kdYsf8PeF_pqDaqEYnIi2mlYALAv4uB9Nn7S0,3239
 zenml/zen_server/routers/projects_endpoints.py,sha256=iIgCg_0_iR5OddgULvsP4xxMOOmEGmtCuSKYZVRE9f4,8464
 zenml/zen_server/routers/run_metadata_endpoints.py,sha256=UTWQvweZTGuZnLrqhbIRYQ-z_TfOPY3Kn9p4YwU0tqw,3662
-zenml/zen_server/routers/run_templates_endpoints.py,sha256=rEREmFmG38O_w6ktVJU5iu-dguGZpSC8RN9qH4WkEsA,8254
+zenml/zen_server/routers/run_templates_endpoints.py,sha256=AdBvwbXefo9nKU2erkNZNFjZ3b8LitPuWR22rgWF3JM,8469
 zenml/zen_server/routers/runs_endpoints.py,sha256=YoTqWm-Mx97q8ZMYXgIY8gr2ICJpRlk6DxxHKN1utFU,12631
 zenml/zen_server/routers/schedule_endpoints.py,sha256=UXt6TIEX0PYhLoBgU-n5kIvWckX1BY_P5uUA_lK9J9A,5782
 zenml/zen_server/routers/secrets_endpoints.py,sha256=diYTC-jl6Hxd_BHVaNsPf2ZcWUwC99naWHsM1xRjR-g,9198
@@ -1064,7 +1064,7 @@ zenml/zen_server/routers/webhook_endpoints.py,sha256=KOJsuykv_TMjL3oEItpC4OWWP75
 zenml/zen_server/secure_headers.py,sha256=glh6QujnjyeoH1_FK-tAS-105G-qKS_34AqSzqJ6TRc,4182
 zenml/zen_server/template_execution/__init__.py,sha256=79knXLKfegsvVSVSWecpqrepq6iAavTUA4hKuiDk-WE,613
 zenml/zen_server/template_execution/runner_entrypoint_configuration.py,sha256=Y8aYJhqqs8Kv8I1q-dM1WemS5VBIfyoaaYH_YkzC7iY,1541
-zenml/zen_server/template_execution/utils.py,sha256=6a62U0hLDe79xLbZM-WbzY-miOeTejQ7vpIkmch0ggU,19086
+zenml/zen_server/template_execution/utils.py,sha256=CiN7d8jCUTV5oKuoNF9Z_eDkitZrhL6V_Tz4CjU7em8,19410
 zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8ajuZE01DaHmdCDCZmsroDcTarvN-nE8jv6qQ,2590
 zenml/zen_server/utils.py,sha256=yOwPI0DxzDIxF7Ppnty_DuGMWYJ_ubHUNWfbTQ1_q9Q,18461
 zenml/zen_server/zen_server_api.py,sha256=1_YHOUmBulXuHWCxDLYtDCMxGze159UmU3f05643PSg,18182
@@ -1315,11 +1315,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=5err1a-TrV3SR5
 zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
-zenml/zen_stores/sql_zen_store.py,sha256=biOoDb2_zYmpsN-J-FSlKICYdwM9KDIe-_KN_yDf_mA,441414
+zenml/zen_stores/sql_zen_store.py,sha256=CK0owrZvLV9u8N0Fa-JrPld-aXxBDncX5MlUexjTIFk,441321
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.82.0.dev20250513.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.82.0.dev20250513.dist-info/METADATA,sha256=hMwBzq4y98Z1jeZUvJMdUimsg1jZUzvmzqDMXpMrSIs,24317
-zenml_nightly-0.82.0.dev20250513.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.82.0.dev20250513.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.82.0.dev20250513.dist-info/RECORD,,
+zenml_nightly-0.82.0.dev20250514.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.82.0.dev20250514.dist-info/METADATA,sha256=t7QND4rpTdtRqG1-abGR9CHeSqTFxKWMoTCLNht1VrQ,24317
+zenml_nightly-0.82.0.dev20250514.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.82.0.dev20250514.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.82.0.dev20250514.dist-info/RECORD,,