zenml-nightly 0.84.1.dev20250805__py3-none-any.whl → 0.84.1.dev20250806__py3-none-any.whl

zenml/integrations/kubernetes/orchestrators/manifest_utils.py

@@ -95,7 +95,7 @@ def add_local_stores_mount(
 
 
 def build_pod_manifest(
-    pod_name: str,
+    pod_name: Optional[str],
     image_name: str,
     command: List[str],
     args: List[str],
@@ -262,90 +262,6 @@ def add_pod_settings(
                 setattr(pod_spec, key, value)
 
 
-def build_cron_job_manifest(
-    cron_expression: str,
-    pod_name: str,
-    image_name: str,
-    command: List[str],
-    args: List[str],
-    privileged: bool,
-    pod_settings: Optional[KubernetesPodSettings] = None,
-    service_account_name: Optional[str] = None,
-    env: Optional[Dict[str, str]] = None,
-    labels: 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,
-    termination_grace_period_seconds: Optional[int] = 30,
-) -> k8s_client.V1CronJob:
-    """Create a manifest for launching a pod as scheduled CRON job.
-
-    Args:
-        cron_expression: CRON job schedule expression, e.g. "* * * * *".
-        pod_name: Name of the pod.
-        image_name: Name of the Docker image.
-        command: Command to execute the entrypoint in the pod.
-        args: Arguments provided to the entrypoint command.
-        privileged: Whether to run the container in privileged mode.
-        pod_settings: Optional settings for the pod.
-        service_account_name: Optional name of a service account.
-            Can be used to assign certain roles to a pod, e.g., to allow it to
-            run Kubernetes commands from within the cluster.
-        env: Environment variables to set.
-        labels: Labels to add to the pod.
-        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.
-        termination_grace_period_seconds: The amount of seconds to wait for a
-            pod to shutdown gracefully.
-
-    Returns:
-        CRON job manifest.
-    """
-    pod_manifest = build_pod_manifest(
-        pod_name=pod_name,
-        image_name=image_name,
-        command=command,
-        args=args,
-        privileged=privileged,
-        pod_settings=pod_settings,
-        service_account_name=service_account_name,
-        env=env,
-        labels=labels,
-        mount_local_stores=mount_local_stores,
-        termination_grace_period_seconds=termination_grace_period_seconds,
-    )
-
-    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(
-                template=k8s_client.V1PodTemplateSpec(
-                    metadata=pod_manifest.metadata,
-                    spec=pod_manifest.spec,
-                ),
-                ttl_seconds_after_finished=ttl_seconds_after_finished,
-            ),
-        ),
-    )
-
-    job_manifest = k8s_client.V1CronJob(
-        kind="CronJob",
-        api_version="batch/v1",
-        metadata=pod_manifest.metadata,
-        spec=job_spec,
-    )
-
-    return job_manifest
-
-
 def build_role_binding_manifest_for_service_account(
     name: str,
     role_name: str,
@@ -475,6 +391,7 @@ def build_job_manifest(
     backoff_limit: Optional[int] = None,
     ttl_seconds_after_finished: Optional[int] = None,
     labels: Optional[Dict[str, str]] = None,
+    annotations: Optional[Dict[str, str]] = None,
     active_deadline_seconds: Optional[int] = None,
     pod_failure_policy: Optional[Dict[str, Any]] = None,
     owner_references: Optional[List[k8s_client.V1OwnerReference]] = None,
@@ -487,6 +404,7 @@ def build_job_manifest(
         backoff_limit: The backoff limit for the job.
         ttl_seconds_after_finished: The TTL seconds after finished for the job.
         labels: The labels to use for the job.
+        annotations: The annotations to use for the job.
         active_deadline_seconds: The active deadline seconds for the job.
         pod_failure_policy: The pod failure policy for the job.
         owner_references: The owner references for the job.
@@ -505,7 +423,57 @@ def build_job_manifest(
     job_metadata = k8s_client.V1ObjectMeta(
         name=job_name,
         labels=labels,
+        annotations=annotations,
         owner_references=owner_references,
     )
 
     return k8s_client.V1Job(spec=job_spec, metadata=job_metadata)
+
+
+def job_template_manifest_from_job(
+    job: k8s_client.V1Job,
+) -> k8s_client.V1JobTemplateSpec:
+    """Build a Kubernetes job template manifest from a job.
+
+    Args:
+        job: The job manifest to build the template from.
+
+    Returns:
+        The job template manifest.
+    """
+    return k8s_client.V1JobTemplateSpec(
+        metadata=job.metadata,
+        spec=job.spec,
+    )
+
+
+def build_cron_job_manifest(
+    job_template: k8s_client.V1JobTemplateSpec,
+    cron_expression: str,
+    successful_jobs_history_limit: Optional[int] = None,
+    failed_jobs_history_limit: Optional[int] = None,
+) -> k8s_client.V1CronJob:
+    """Build a Kubernetes cron job manifest.
+
+    Args:
+        job_template: The job template to use for the cron job.
+        cron_expression: The cron expression to use for the cron job.
+        successful_jobs_history_limit: The number of successful jobs to keep.
+        failed_jobs_history_limit: The number of failed jobs to keep.
+
+    Returns:
+        The Kubernetes cron job manifest.
+    """
+    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=job_template,
+    )
+
+    return k8s_client.V1CronJob(
+        kind="CronJob",
+        api_version="batch/v1",
+        metadata=job_template.metadata,
+        spec=spec,
+    )

zenml/integrations/kubernetes/step_operators/kubernetes_step_operator.py

@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 """Kubernetes step operator implementation."""
 
+import random
 from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Type, cast
 
 from kubernetes import client as k8s_client
@@ -20,13 +21,21 @@ from kubernetes import client as k8s_client
 from zenml.config.base_settings import BaseSettings
 from zenml.config.build_configuration import BuildConfiguration
 from zenml.enums import StackComponentType
+from zenml.integrations.kubernetes.constants import (
+    STEP_NAME_ANNOTATION_KEY,
+    STEP_OPERATOR_ANNOTATION_KEY,
+)
 from zenml.integrations.kubernetes.flavors import (
     KubernetesStepOperatorConfig,
     KubernetesStepOperatorSettings,
 )
-from zenml.integrations.kubernetes.orchestrators import kube_utils
+from zenml.integrations.kubernetes.orchestrators import (
+    kube_utils,
+)
 from zenml.integrations.kubernetes.orchestrators.manifest_utils import (
+    build_job_manifest,
     build_pod_manifest,
+    pod_template_manifest_from_pod,
 )
 from zenml.logger import get_logger
 from zenml.stack import Stack, StackValidator
@@ -174,6 +183,15 @@ class KubernetesStepOperator(BaseStepOperator):
         """
         return k8s_client.CoreV1Api(self.get_kube_client())
 
+    @property
+    def _k8s_batch_api(self) -> k8s_client.BatchV1Api:
+        """Getter for the Kubernetes Batch API client.
+
+        Returns:
+            The Kubernetes Batch API client.
+        """
+        return k8s_client.BatchV1Api(self.get_kube_client())
+
     def launch(
         self,
         info: "StepRunInfo",
@@ -194,53 +212,77 @@ class KubernetesStepOperator(BaseStepOperator):
         image_name = info.get_image(
             key=KUBERNETES_STEP_OPERATOR_DOCKER_IMAGE_KEY
         )
-
-        pod_name = f"{info.run_name}_{info.pipeline_step_name}"
-        pod_name = kube_utils.sanitize_pod_name(
-            pod_name, namespace=self.config.kubernetes_namespace
-        )
-
         command = entrypoint_command[:3]
         args = entrypoint_command[3:]
 
-        # Create and run the orchestrator pod.
+        step_labels = {
+            "run_id": kube_utils.sanitize_label(str(info.run_id)),
+            "run_name": kube_utils.sanitize_label(str(info.run_name)),
+            "pipeline": kube_utils.sanitize_label(info.pipeline.name),
+            "step_name": kube_utils.sanitize_label(info.pipeline_step_name),
+        }
+        step_annotations = {
+            STEP_NAME_ANNOTATION_KEY: info.pipeline_step_name,
+            STEP_OPERATOR_ANNOTATION_KEY: str(self.id),
+        }
+
+        # 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
+        # scheduled on any node regardless of available memory and risk
+        # negatively impacting or even crashing the node due to memory pressure.
+        pod_settings = kube_utils.apply_default_resource_requests(
+            memory="400Mi",
+            pod_settings=settings.pod_settings,
+        )
+
         pod_manifest = build_pod_manifest(
-            pod_name=pod_name,
+            pod_name=None,
             image_name=image_name,
             command=command,
             args=args,
+            env=environment,
             privileged=settings.privileged,
+            pod_settings=pod_settings,
             service_account_name=settings.service_account_name,
-            pod_settings=settings.pod_settings,
-            env=environment,
-            mount_local_stores=False,
-            labels={
-                "run_id": kube_utils.sanitize_label(str(info.run_id)),
-                "pipeline": kube_utils.sanitize_label(info.pipeline.name),
-            },
+            labels=step_labels,
         )
 
-        kube_utils.create_and_wait_for_pod_to_start(
-            core_api=self._k8s_core_api,
-            pod_display_name=f"pod of step `{info.pipeline_step_name}`",
-            pod_name=pod_name,
-            pod_manifest=pod_manifest,
+        job_name = settings.job_name_prefix or ""
+        random_prefix = "".join(random.choices("0123456789abcdef", k=8))
+        job_name += f"-{random_prefix}-{info.pipeline_step_name}-{info.pipeline.name}-step-operator"
+        # The job name will be used as a label on the pods, so we need to make
+        # sure it doesn't exceed the label length limit
+        job_name = kube_utils.sanitize_label(job_name)
+
+        job_manifest = build_job_manifest(
+            job_name=job_name,
+            pod_template=pod_template_manifest_from_pod(pod_manifest),
+            # The orchestrator already handles retries, so we don't need to
+            # retry the step operator job.
+            backoff_limit=0,
+            ttl_seconds_after_finished=settings.ttl_seconds_after_finished,
+            active_deadline_seconds=settings.active_deadline_seconds,
+            labels=step_labels,
+            annotations=step_annotations,
+        )
+
+        kube_utils.create_job(
+            batch_api=self._k8s_batch_api,
             namespace=self.config.kubernetes_namespace,
-            startup_max_retries=settings.pod_failure_max_retries,
-            startup_failure_delay=settings.pod_failure_retry_delay,
-            startup_failure_backoff=settings.pod_failure_backoff,
-            startup_timeout=settings.pod_startup_timeout,
+            job_manifest=job_manifest,
         )
 
         logger.info(
-            "Waiting for pod of step `%s` to finish...",
-            info.pipeline_step_name,
+            "Waiting for step operator job `%s` to finish...",
+            job_name,
         )
-        kube_utils.wait_pod(
-            kube_client_fn=self.get_kube_client,
-            pod_name=pod_name,
+        kube_utils.wait_for_job_to_finish(
+            batch_api=self._k8s_batch_api,
+            core_api=self._k8s_core_api,
             namespace=self.config.kubernetes_namespace,
-            exit_condition_lambda=kube_utils.pod_is_done,
+            job_name=job_name,
+            fail_on_container_waiting_reasons=settings.fail_on_container_waiting_reasons,
             stream_logs=True,
         )
-        logger.info("Pod of step `%s` completed.", info.pipeline_step_name)
+        logger.info("Step operator job completed.")

zenml/logging/step_logging.py

@@ -48,6 +48,7 @@ from zenml.exceptions import DoesNotExistException
 from zenml.logger import get_logger
 from zenml.models import (
     LogsRequest,
+    LogsResponse,
     PipelineDeploymentResponse,
     PipelineRunUpdate,
 )
@@ -702,7 +703,9 @@ class PipelineLogsStorageContext:
 
 
 def setup_orchestrator_logging(
-    run_id: str, deployment: "PipelineDeploymentResponse"
+    run_id: UUID,
+    deployment: "PipelineDeploymentResponse",
+    logs_response: Optional[LogsResponse] = None,
 ) -> Any:
     """Set up logging for an orchestrator environment.
 
@@ -712,61 +715,61 @@ def setup_orchestrator_logging(
     Args:
         run_id: The pipeline run ID.
         deployment: The deployment of the pipeline run.
+        logs_response: The logs response to continue from.
 
     Returns:
         The logs context (PipelineLogsStorageContext)
     """
     try:
-        step_logging_enabled = True
+        logging_enabled = True
 
-        # Check whether logging is enabled
         if handle_bool_env_var(ENV_ZENML_DISABLE_PIPELINE_LOGS_STORAGE, False):
-            step_logging_enabled = False
+            logging_enabled = False
         else:
             if (
                 deployment.pipeline_configuration.enable_pipeline_logs
                 is not None
             ):
-                step_logging_enabled = (
+                logging_enabled = (
                     deployment.pipeline_configuration.enable_pipeline_logs
                 )
 
-        if not step_logging_enabled:
+        if not logging_enabled:
             return nullcontext()
 
         # Fetch the active stack
         client = Client()
         active_stack = client.active_stack
 
-        # Configure the logs
-        logs_uri = prepare_logs_uri(
-            artifact_store=active_stack.artifact_store,
-        )
+        if logs_response:
+            logs_uri = logs_response.uri
+        else:
+            logs_uri = prepare_logs_uri(
+                artifact_store=active_stack.artifact_store,
+            )
+            logs_model = LogsRequest(
+                uri=logs_uri,
+                source="orchestrator",
+                artifact_store_id=active_stack.artifact_store.id,
+            )
+
+            # Add orchestrator logs to the pipeline run
+            try:
+                run_update = PipelineRunUpdate(add_logs=[logs_model])
+                client.zen_store.update_run(
+                    run_id=run_id, run_update=run_update
+                )
+            except Exception as e:
+                logger.error(
+                    f"Failed to add orchestrator logs to the run {run_id}: {e}"
+                )
+                raise e
 
-        logs_context = PipelineLogsStorageContext(
+        return PipelineLogsStorageContext(
             logs_uri=logs_uri,
             artifact_store=active_stack.artifact_store,
             prepend_step_name=False,
         )
-
-        logs_model = LogsRequest(
-            uri=logs_uri,
-            source="orchestrator",
-            artifact_store_id=active_stack.artifact_store.id,
-        )
-
-        # Add orchestrator logs to the pipeline run
-        try:
-            run_update = PipelineRunUpdate(add_logs=[logs_model])
-            client.zen_store.update_run(
-                run_id=UUID(run_id), run_update=run_update
-            )
-        except Exception as e:
-            logger.error(
-                f"Failed to add orchestrator logs to the run {run_id}: {e}"
-            )
-            raise e
-        return logs_context
     except Exception as e:
         logger.error(
             f"Failed to setup orchestrator logging for run {run_id}: {e}"

zenml/steps/base_step.py

@@ -104,8 +104,8 @@ class BaseStep:
         enable_artifact_metadata: Optional[bool] = None,
         enable_artifact_visualization: Optional[bool] = None,
         enable_step_logs: Optional[bool] = None,
-        experiment_tracker: Optional[str] = None,
-        step_operator: Optional[str] = None,
+        experiment_tracker: Optional[Union[bool, str]] = None,
+        step_operator: Optional[Union[bool, str]] = None,
         parameters: Optional[Dict[str, Any]] = None,
         output_materializers: Optional[
             "OutputMaterializersSpecification"
@@ -594,8 +594,8 @@ class BaseStep:
         enable_artifact_metadata: Optional[bool] = None,
         enable_artifact_visualization: Optional[bool] = None,
         enable_step_logs: Optional[bool] = None,
-        experiment_tracker: Optional[str] = None,
-        step_operator: Optional[str] = None,
+        experiment_tracker: Optional[Union[bool, str]] = None,
+        step_operator: Optional[Union[bool, str]] = None,
         parameters: Optional[Dict[str, Any]] = None,
         output_materializers: Optional[
             "OutputMaterializersSpecification"
@@ -728,8 +728,8 @@ class BaseStep:
         enable_artifact_metadata: Optional[bool] = None,
         enable_artifact_visualization: Optional[bool] = None,
         enable_step_logs: Optional[bool] = None,
-        experiment_tracker: Optional[str] = None,
-        step_operator: Optional[str] = None,
+        experiment_tracker: Optional[Union[bool, str]] = None,
+        step_operator: Optional[Union[bool, str]] = None,
         parameters: Optional[Dict[str, Any]] = None,
         output_materializers: Optional[
             "OutputMaterializersSpecification"

zenml/steps/step_decorator.py

@@ -64,8 +64,8 @@ def step(
     enable_artifact_metadata: Optional[bool] = None,
     enable_artifact_visualization: Optional[bool] = None,
     enable_step_logs: Optional[bool] = None,
-    experiment_tracker: Optional[str] = None,
-    step_operator: Optional[str] = None,
+    experiment_tracker: Optional[Union[bool, str]] = None,
+    step_operator: Optional[Union[bool, str]] = None,
     output_materializers: Optional["OutputMaterializersSpecification"] = None,
     settings: Optional[Dict[str, "SettingsOrDict"]] = None,
     extra: Optional[Dict[str, Any]] = None,
@@ -85,8 +85,8 @@ def step(
     enable_artifact_metadata: Optional[bool] = None,
     enable_artifact_visualization: Optional[bool] = None,
     enable_step_logs: Optional[bool] = None,
-    experiment_tracker: Optional[str] = None,
-    step_operator: Optional[str] = None,
+    experiment_tracker: Optional[Union[bool, str]] = None,
+    step_operator: Optional[Union[bool, str]] = None,
     output_materializers: Optional["OutputMaterializersSpecification"] = None,
     settings: Optional[Dict[str, "SettingsOrDict"]] = None,
     extra: Optional[Dict[str, Any]] = None,

{zenml_nightly-0.84.1.dev20250805.dist-info → zenml_nightly-0.84.1.dev20250806.dist-info}/METADATA

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

{zenml_nightly-0.84.1.dev20250805.dist-info → zenml_nightly-0.84.1.dev20250806.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=JgD203b6qhEP0gZ0HZYxfmgZpi3b8WAfHxs7VQjEn5I,19
+zenml/VERSION,sha256=KMpAeP_9YhseXa7vHadCSXfPfoKYBJW_PucPxE3onyI,19
 zenml/__init__.py,sha256=r7JUg2SVDf_dPhS7iU6vudKusEqK4ics7_jFMZhq0o4,2731
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -333,21 +333,23 @@ zenml/integrations/kubeflow/orchestrators/__init__.py,sha256=J879DBt9WbpojBTdOXy
 zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=lPQSQn6Jp2oJxULG_G8UcbE_bgw_9XTlAjdsujahWbY,38195
 zenml/integrations/kubeflow/orchestrators/local_deployment_utils.py,sha256=qszoOdvBpgIp40XkncphXAr9dRKnyZzGiz2mJ56bYmw,15448
 zenml/integrations/kubernetes/__init__.py,sha256=k1bfrdI1u5RBnAh7yT4w-m-4SWgZ7b4L5uu7dPRc0SI,1809
+zenml/integrations/kubernetes/constants.py,sha256=7pejIdiOAwcQNkWOkqBta_FFMllzsKte5l2fcFdaqN8,1232
 zenml/integrations/kubernetes/flavors/__init__.py,sha256=a5gU45qCj3FkLwl_uVjlIkL_2F5DHk-w1gdcZrvVjBI,1266
-zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=zhcn5d3N4VmlVq-cxtyXtp63AKcwVfKWg-jAlkx2B10,14420
-zenml/integrations/kubernetes/flavors/kubernetes_step_operator_flavor.py,sha256=Vp7kY9t2TWdWRBNSf_sJKPL5dZnhuHnskPhQ7KBuGPY,6384
+zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=O_fnKZ6jPItdOyhgxCTJTD7gpIx1MHhU9_22Cg5x0Ig,15266
+zenml/integrations/kubernetes/flavors/kubernetes_step_operator_flavor.py,sha256=unk8I8uYXn3pJS3HajEhZodhsdFvvhCTs5NzzqkQk3Q,7453
 zenml/integrations/kubernetes/orchestrators/__init__.py,sha256=TJID3OTieZBox36WpQpzD0jdVRA_aZVcs_bNtfXS8ik,811
-zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=Gbw1mT0z-F_t2iQKPmLntGhdNTT-CZNHO5sDb-XxE68,27709
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=Agx_Thj6nByAd5p61HTf0p6SRDPOcYKSSl-_Km2EdfU,41316
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=jgxnJdBVFVn6l6gSoygIoScQxyKh5cVbgXF10qFA9Nk,20809
+zenml/integrations/kubernetes/orchestrators/dag_runner.py,sha256=IDumuhi9hySiTZ7oTI6hz5RlJ4kiuv3NpVYgA_pLS_Y,12044
+zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=soQPph_FWFPAwL7vpZX91cu8jEjo4xBYnafLUdMNtnw,38684
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=TnrKa6xnrYtAq1elyob0aRy-aTLrSjLuuOzdLSs_GFs,36811
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=QahXJ-cM6lrXqWs2BHMQtKQf3svqKB5l69d4D0OB4qc,24789
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=QOwQnWCfB-t_BQ2eOZN0SakurGUd0GTMCSdUlREhk6I,2324
-zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=52Ba6MZ4lCo6CpnW0IjnFOJQNpPoVgbvIqR-x32ubew,16085
+zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=JskjDSbR3HdTxGT3NOVi0ha7Ku3CqMkPzMJJq4IZYp0,14491
 zenml/integrations/kubernetes/pod_settings.py,sha256=s6I0sALXB7PnRqJ-nlU71o3cTm2GM1TfqsE44qhiDJY,8022
 zenml/integrations/kubernetes/serialization_utils.py,sha256=Uy30YWUU8VAizYLWk0cZKtEisSo4AzuzsyGU_KIb8PY,7044
 zenml/integrations/kubernetes/service_connectors/__init__.py,sha256=Uf6zlHIapYrRDl3xOPWQ2jA7jt85SXx1U7DmSxzxTvQ,818
 zenml/integrations/kubernetes/service_connectors/kubernetes_service_connector.py,sha256=Cv4tiVxoQOz9ex0lf3JdJrooEkgMwfDfwt5GOeNRpQU,19669
 zenml/integrations/kubernetes/step_operators/__init__.py,sha256=40utDPYAezxHsFgO0UUIT_6XpCDzDapje6OH951XsTs,806
-zenml/integrations/kubernetes/step_operators/kubernetes_step_operator.py,sha256=__7t6XuzedUEheUImSkX2grsDCz0YFWUJ6uxv47MXUo,8851
+zenml/integrations/kubernetes/step_operators/kubernetes_step_operator.py,sha256=yNCBOwjF9zN-UeI2jTEz6zMmmtx0xiwTdq5Z1JH-Yhg,10506
 zenml/integrations/label_studio/__init__.py,sha256=sF2c9FxTDRlbcu95OxaUNKNtIhC1LgfmBRKY4jBME38,1475
 zenml/integrations/label_studio/annotators/__init__.py,sha256=YtOtSfS1_NBoLoXIygEerElBP1-B98UU0HOAEfzdRY0,821
 zenml/integrations/label_studio/annotators/label_studio_annotator.py,sha256=VkuW4zsZhHz8__P9WTTLRTF-FOmoYB-_cFqBdu-PUyA,30498
@@ -578,7 +580,7 @@ zenml/io/filesystem_registry.py,sha256=stujDg4a5k983WMwp3rj4Z4puiUco4REyVoIoMIpI
 zenml/io/local_filesystem.py,sha256=xQTZPT5cpooptUV8KiifxZojS6pWCv1-6UUxitUYb_E,7386
 zenml/logger.py,sha256=STfebaOdCGmw404gwKhwpDrrhtKDUREeerO7UtapzfE,6997
 zenml/logging/__init__.py,sha256=knhroJ2h0uHBCGzAiBBGJEiuhEA3cwI6XYBRIyXdbkQ,613
-zenml/logging/step_logging.py,sha256=PTAHlqNfWF-Ly0zvBxroQ3S9vOMEhjSZTzi0cmZVKqQ,27767
+zenml/logging/step_logging.py,sha256=dv_lo4rNdBCHsCaYOBoVwKAorY0Rmxh_6AhcGy_Y78g,27922
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
 zenml/login/credentials.py,sha256=RtLmYkFQ5trbprhsO8NCRKwBA136KzGoUWY52dZP9GA,12722
 zenml/login/credentials_store.py,sha256=k55mNSqc_RaBuvLWtg5ubQAwlSbaTxTVAqoZ1OQ2YcM,24750
@@ -754,11 +756,11 @@ zenml/step_operators/__init__.py,sha256=tqj7fgnQyZubLjwUu4ITwkA-70KMQz4g37agbfF7
 zenml/step_operators/base_step_operator.py,sha256=ZRnY6lcEHY8xZskrKKdPhgKg2BlEoh2_kb8xCaM2D1g,3522
 zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=wuQKmEI_ckn1CHHrxWGGdIhQyBFXG01aKC2-2b6Atjo,3690
 zenml/steps/__init__.py,sha256=KKWFOmCZGLDEikOD2E5YmDA7QHo47uPV37by21WwI0U,1453
-zenml/steps/base_step.py,sha256=qzW9I99PWK8brm3wdV1NEuUxGFlMvrmqNh3YjNKUkgg,44312
+zenml/steps/base_step.py,sha256=6mz8ZBn4wd509KcwZeD1qdxQV4BF3wLknMMzn_g3wgg,44390
 zenml/steps/decorated_step.py,sha256=C8Ng5PCLc9eql4JF1N345HQ6LyC1qCUdTnysUTeoAJs,1315
 zenml/steps/entrypoint_function_utils.py,sha256=H0WIkHpR_R0S9gl_tWr0nX-fcBlYnm8OQ1cimvrw-qo,9555
 zenml/steps/step_context.py,sha256=sFvVVvEyKmWTrucofor8Cuxv-72jbziVaQY-OlOvFAM,15526
-zenml/steps/step_decorator.py,sha256=cbu7s2EZ6-yIgBlY6npour2X_CnUCkBfcXfJoiSb3iQ,6454
+zenml/steps/step_decorator.py,sha256=0TKxP_oAVRizjN3JhBZ8ShFd3fSBdVPU2ijytTg1h2U,6506
 zenml/steps/step_invocation.py,sha256=ETfOaV-P4_iXGk9y1-xK54Kfg2QRYaGoj_jTyEYZfb0,4861
 zenml/steps/utils.py,sha256=UpZ5lN46Bqu1eXyVfh969M4-UBbOZjq5R7u4PkXYDZ8,18586
 zenml/types.py,sha256=LhGWJ4t3nybBk1l9Ox3tqqHbTYSuCqhkRsL5FqO6yf4,1206
@@ -1352,8 +1354,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=-3zeByIUjIvsx3564O2gJ463512QkZl04okL3eB-nJs,491568
 zenml/zen_stores/template_utils.py,sha256=iCXrXpqzVTY7roqop4Eh9J7DmLW6PQeILZexmw_l3b8,10074
 zenml/zen_stores/zen_store_interface.py,sha256=weiSULdI9AsbCE10a5TcwtybX-BJs9hKhjPJnTapWv4,93023
-zenml_nightly-0.84.1.dev20250805.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.84.1.dev20250805.dist-info/METADATA,sha256=jJ1Pq_B-npNNW5qFIT8aOon1RsIHxqo2E7EIUSdPF6U,24296
-zenml_nightly-0.84.1.dev20250805.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.84.1.dev20250805.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.84.1.dev20250805.dist-info/RECORD,,
+zenml_nightly-0.84.1.dev20250806.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.84.1.dev20250806.dist-info/METADATA,sha256=VZaHyV--b2EcSDdhFPJE0noToZKb34x6Cq6CKNDsSk4,24296
+zenml_nightly-0.84.1.dev20250806.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.84.1.dev20250806.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.84.1.dev20250806.dist-info/RECORD,,