zenml-nightly 0.83.1.dev20250702__py3-none-any.whl → 0.83.1.dev20250704__py3-none-any.whl

zenml/zen_server/routers/runs_endpoints.py

@@ -25,6 +25,7 @@ from zenml.constants import (
     RUNS,
     STATUS,
     STEPS,
+    STOP,
     VERSION_1,
 )
 from zenml.enums import ExecutionStatus, StackComponentType
@@ -40,6 +41,7 @@ from zenml.models import (
     StepRunFilter,
     StepRunResponse,
 )
+from zenml.utils import run_utils
 from zenml.zen_server.auth import AuthContext, authorize
 from zenml.zen_server.exceptions import error_response
 from zenml.zen_server.rbac.endpoint_utils import (
@@ -51,6 +53,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 )
 from zenml.zen_server.rbac.models import Action, ResourceType
 from zenml.zen_server.rbac.utils import (
+    dehydrate_response_model,
     verify_permission_for_model,
 )
 from zenml.zen_server.routers.projects_endpoints import workspace_router
@@ -389,38 +392,39 @@ def refresh_run_status(
 
     Args:
         run_id: ID of the pipeline run to refresh.
-
-    Raises:
-        RuntimeError: If the stack or the orchestrator of the run is deleted.
     """
-    # Verify access to the run
     run = verify_permissions_and_get_entity(
         id=run_id,
         get_method=zen_store().get_run,
         hydrate=True,
     )
-
-    # Check the stack and its orchestrator
-    if run.stack is not None:
-        orchestrators = run.stack.components.get(
-            StackComponentType.ORCHESTRATOR, []
-        )
-        if orchestrators:
-            verify_permission_for_model(
-                model=orchestrators[0], action=Action.READ
-            )
-        else:
-            raise RuntimeError(
-                f"The orchestrator, the run '{run.id}' was executed with, is "
-                "deleted."
-            )
-    else:
-        raise RuntimeError(
-            f"The stack, the run '{run.id}' was executed on, is deleted."
-        )
     run.refresh_run_status()
 
 
+@router.post(
+    "/{run_id}" + STOP,
+    responses={401: error_response, 404: error_response, 422: error_response},
+)
+@async_fastapi_endpoint_wrapper
+def stop_run(
+    run_id: UUID,
+    graceful: bool = False,
+    _: AuthContext = Security(authorize),
+) -> None:
+    """Stops a specific pipeline run.
+
+    Args:
+        run_id: ID of the pipeline run to stop.
+        graceful: If True, allows for graceful shutdown where possible.
+            If False, forces immediate termination. Default is False.
+    """
+    run = zen_store().get_run(run_id, hydrate=True)
+    verify_permission_for_model(run, action=Action.READ)
+    verify_permission_for_model(run, action=Action.UPDATE)
+    dehydrate_response_model(run)
+    run_utils.stop_run(run=run, graceful=graceful)
+
+
 @router.get(
     "/{run_id}/logs",
     responses={

zenml/zen_stores/sql_zen_store.py

@@ -6119,7 +6119,14 @@ class SqlZenStore(BaseZenStore):
                 resources=existing_run,
                 session=session,
             )
+
+            if run_update.status is not None:
+                self._update_pipeline_run_status(
+                    pipeline_run_id=run_id,
+                    session=session,
+                )
             session.refresh(existing_run)
+
             return existing_run.to_model(
                 include_metadata=True, include_resources=True
             )
@@ -8824,6 +8831,7 @@ class SqlZenStore(BaseZenStore):
 
         Raises:
             EntityExistsError: if the step run already exists.
+            IllegalOperationError: if the pipeline run is stopped or stopping.
         """
         with Session(self.engine) as session:
             self._set_request_user_id(request_model=step_run, session=session)
@@ -8835,6 +8843,16 @@ class SqlZenStore(BaseZenStore):
                 reference_id=step_run.pipeline_run_id,
                 session=session,
             )
+
+            # Validate pipeline status before creating step
+            if run.status in [
+                ExecutionStatus.STOPPING,
+                ExecutionStatus.STOPPED,
+            ]:
+                raise IllegalOperationError(
+                    f"Cannot create step '{step_run.name}' for pipeline in "
+                    f"{run.status} state. Pipeline run ID: {step_run.pipeline_run_id}"
+                )
             self._get_reference_schema_by_id(
                 resource=step_run,
                 reference_schema=StepRunSchema,
@@ -8996,6 +9014,8 @@ class SqlZenStore(BaseZenStore):
                         session=session,
                     )
 
+            session.commit()
+
             if step_run.status != ExecutionStatus.RUNNING:
                 self._update_pipeline_run_status(
                     pipeline_run_id=step_run.pipeline_run_id, session=session
@@ -9130,15 +9150,14 @@ class SqlZenStore(BaseZenStore):
                     input_type=StepRunInputArtifactType.MANUAL,
                     session=session,
                 )
+            session.commit()
+            session.refresh(existing_step_run)
 
             self._update_pipeline_run_status(
                 pipeline_run_id=existing_step_run.pipeline_run_id,
                 session=session,
             )
 
-            session.commit()
-            session.refresh(existing_step_run)
-
             return existing_step_run.to_model(
                 include_metadata=True, include_resources=True
             )
@@ -9375,6 +9394,7 @@ class SqlZenStore(BaseZenStore):
         assert pipeline_run.deployment
         num_steps = pipeline_run.deployment.step_count
         new_status = get_pipeline_run_status(
+            run_status=ExecutionStatus(pipeline_run.status),
             step_statuses=[
                 ExecutionStatus(status) for status in step_run_statuses
             ],

{zenml_nightly-0.83.1.dev20250702.dist-info → zenml_nightly-0.83.1.dev20250704.dist-info}/METADATA

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

{zenml_nightly-0.83.1.dev20250702.dist-info → zenml_nightly-0.83.1.dev20250704.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=IfWa1bmv2tmdzTDgYQqFV4nSQFI19iQ6CS0IzuNeNI8,19
+zenml/VERSION,sha256=ZKa34RNVKLuDoFihHGmwjsp2-bfKkOT7ax788HZQsUw,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
@@ -41,7 +41,7 @@ zenml/cli/integration.py,sha256=Rq37R-QiSvaFOMmlA0YJTwDti4slLp-hEf05vbWxkFk,1734
 zenml/cli/login.py,sha256=bS2hhJAU4FtnNZ9Y3M_q-q9_ga_UnkfBK88DD0A3Ook,39509
 zenml/cli/model.py,sha256=_6la1czUXPBWBkiZLS6iwGHMy2z4vJLX0Z2Ho2zV48M,21921
 zenml/cli/model_registry.py,sha256=zzxWXXFhKu2B1Wp0u7prKVnN1ftM-JdGdQwlD-5G-QM,20786
-zenml/cli/pipeline.py,sha256=Vlz1OgGb1Ep-4Ekgd-Wz5SmieWigfx56i8wA5BGl228,19222
+zenml/cli/pipeline.py,sha256=ulmIzHeV_DJpYhJBWP535iSpShINManuAwy4LGeSPsE,20908
 zenml/cli/project.py,sha256=oo9rxjwcX9Mi-0ZtiTMOoajQ8JrEkl23BcNFZxjfhj0,6536
 zenml/cli/secret.py,sha256=zwt07v0xoIf_dLf-qY5CFdbKepBEuwmXD2HIMcLe_xU,20164
 zenml/cli/served_model.py,sha256=3w1UcAbg6Geu37fr7ej1_81GBCt3fF7j3Ge799YE4Mc,14974
@@ -53,7 +53,7 @@ zenml/cli/stack_components.py,sha256=QqakqWsvzgG7nOvQmvlbcfwDZUTsXI2pOlsOBXP5EXg
 zenml/cli/tag.py,sha256=JiAoYyDDBBYj0ChT8zWY8KGjzAnJTNx-WLQrjLLEyik,4802
 zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=sNnhaUxH-cHecbZBR1L0mEU0TnLNZHzI6ZBCUSQa7OY,13078
-zenml/cli/utils.py,sha256=6Ld_CK4k5B-Kn8ilGFiV-J4Xl8JEX36PZrNFIT5slLo,85310
+zenml/cli/utils.py,sha256=pppn7IJWMhWv978l6W2mV0AxX5-UKsA6YZp69cczYKQ,85419
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
 zenml/client.py,sha256=fLYjWtQEqvJqB-IV-WQ-54bKNRSRtlU4XljMxydvN1c,293403
 zenml/client_lazy_loader.py,sha256=oyxKvBWVB7k2pHMavdhNEOfR2Vk4IS3XUu43SBzDPsI,7152
@@ -65,7 +65,7 @@ zenml/code_repositories/local_repository_context.py,sha256=1VyiYkJBDVg0iGusgRQDT
 zenml/config/__init__.py,sha256=NIMS7QQo3-uVkNlzPvR5FbtgPdRibF2xFJB7Vr0fhCM,1826
 zenml/config/base_settings.py,sha256=itoLqc1cOwEYhgSGdZmSKSaBevQkvYH7NQh7PUamazc,1700
 zenml/config/build_configuration.py,sha256=jGGNwP0Cb7a80JXArNxDgxzxl9ytSZRtv-WW7_psLbM,6870
-zenml/config/compiler.py,sha256=dVQ2FfliNxt93H2SYwlSOVyh-5dmdMd7abbIVRuIn3I,23023
+zenml/config/compiler.py,sha256=xJoJFDZ-G1M5Lqq8O7dHjm3e3KcV8Iaefuf-rc4I7QI,23794
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=xZvoeC6v6RG_5MFK_RgdTgrkOLu-QT6frY4SGQNZUTo,18460
 zenml/config/global_config.py,sha256=59orl6xp4moxSo2Q8tXtDfpIBX4AxJ9FHcm8s2VHyWw,29712
@@ -80,12 +80,12 @@ zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wz
 zenml/config/server_config.py,sha256=HNHODfojAstBWMxcV5OjxcXSh8Oy9k-_0J5la60fqzQ,32616
 zenml/config/settings_resolver.py,sha256=PR9BRm_x1dy7nVKa9UqpeFdck8IEATSW6aWT8FKd-DI,4278
 zenml/config/source.py,sha256=RzUw8lin8QztUjz-AdoCzVM5Om_cSSPuroaPx-qAO4w,8226
-zenml/config/step_configurations.py,sha256=5iizSZNU0d2dpeHZnr0ODpaZ1Fxc2x5e3wa9uOmQzlY,13614
+zenml/config/step_configurations.py,sha256=fcUDGezCdbUYkz2uHuXsuSJTtge_V7XNUvZTRLAiJO8,14651
 zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I,1888
 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=DSMrlFcC2dWXlGOOEMoeK28qnVj9eKWjMBj39bGAWPU,16716
+zenml/constants.py,sha256=8YAgT4Y9pwqsNxnplpdCti7_IsomsSgdqOVG9tPnWJI,16731
 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
@@ -100,7 +100,7 @@ zenml/entrypoints/base_entrypoint_configuration.py,sha256=t7sU2MEEPVomX9dZCpPhk1
 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=hdw54YLwHXyFREySih3F8BTRavj_QeS_hAsgrnU-gCc,7790
-zenml/enums.py,sha256=HUpgdpf6mDmTieeTFb_K03HZRmCWDQRwHv7xrHcKc1E,11391
+zenml/enums.py,sha256=cLERtpwFX3RI8vQPvVhMWkYrayKNqzMXB-lcbnjsflQ,11461
 zenml/environment.py,sha256=_iS9o4rTxRCjLocSw_LEe9pZ9jfyMhXBY47LIUF3VCI,12255
 zenml/event_hub/__init__.py,sha256=-fD9mPOslf4J-_XFBPp5gYlPz7-6ZaAKHa5jxf_nyAo,757
 zenml/event_hub/base_event_hub.py,sha256=PqKrnvOye0UUS3s09zGgjI5dtj0IwzEBDbavA_PgfZQ,6579
@@ -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=YIOFic8CpPaBrQr28K6d2gBNeZRKaDArya3cGWS9a-s,7327
+zenml/exceptions.py,sha256=iDnsJotbJgBWmmJ9XNDwVLquvP6AMWgHAWTPshbOFnQ,7573
 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
@@ -146,12 +146,12 @@ zenml/integrations/aws/flavors/sagemaker_step_operator_flavor.py,sha256=e3locb2O
 zenml/integrations/aws/image_builders/__init__.py,sha256=91hgH1OphG2i-vk-G8N4yKBFIzK89Wu7BK4-T5yOA7E,786
 zenml/integrations/aws/image_builders/aws_image_builder.py,sha256=UcPYYYjJjfsicY3hV4OZeJt552AVmwPZPlv-AsG1g1I,11489
 zenml/integrations/aws/orchestrators/__init__.py,sha256=Wh0Fhtt_uo6YrkvXY9kL0M478FL7XpapjoFreUZbgUg,794
-zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py,sha256=kX_fPcjrxrNe0xiIgNgfT0ypEycAVjNkqj4iL-hPQOY,39289
+zenml/integrations/aws/orchestrators/sagemaker_orchestrator.py,sha256=buj13k-7jvmAsA19c6JYcjCPrjtckgvT-YrgisXNahw,39432
 zenml/integrations/aws/orchestrators/sagemaker_orchestrator_entrypoint_config.py,sha256=WXCWdVojIZxx5_3-g1T95S2vsJ-RLNGcp-V409wgme0,1555
 zenml/integrations/aws/service_connectors/__init__.py,sha256=w2Md40yG89PwmU9eBceh6dGy3XYZ3MKusNAZ51sGOgE,783
 zenml/integrations/aws/service_connectors/aws_service_connector.py,sha256=7H69IoOYmyn5QcXEfL1-OmC0UaQ54TfNNhv2t8A6Daw,92107
 zenml/integrations/aws/step_operators/__init__.py,sha256=HK5oIqLixYKUKoiN4LnqTyGjAZJUdUqWYGqJhFSWyo0,817
-zenml/integrations/aws/step_operators/sagemaker_step_operator.py,sha256=1RCHSzHcgTldChlk4c8iCrKTeeHrdQlQUJpMepk9ShE,10384
+zenml/integrations/aws/step_operators/sagemaker_step_operator.py,sha256=16y1av9ws_4FEGxxFqNSXOJXHDZilbgNWwBhv_ZRur4,10387
 zenml/integrations/aws/step_operators/sagemaker_step_operator_entrypoint_config.py,sha256=2cXroe6-bXyHVkO5yPnZagNpqx6MrMDcvuvNr8J2j-A,1581
 zenml/integrations/azure/__init__.py,sha256=iJ8smVn76fIuOeeJV5txrBfBS0PyRKUNt9aVkL3Ucic,3103
 zenml/integrations/azure/artifact_stores/__init__.py,sha256=dlIwbpgjE0Hy4rhMbelNJHVKm4t8tj_hRu9mQ_cEIAg,820
@@ -163,12 +163,12 @@ zenml/integrations/azure/flavors/azureml.py,sha256=S40e2vlKcpFZZJ5NkvUMX5HQAOWdx
 zenml/integrations/azure/flavors/azureml_orchestrator_flavor.py,sha256=XR_UI-Bvs7W1S_lFsGKqEiIjhVL87sY3_VOcC9WsrTY,4965
 zenml/integrations/azure/flavors/azureml_step_operator_flavor.py,sha256=jBA8WuChZ8hJy_xumG6pf-2HTxiRznJ6YRmi8KePD_I,6267
 zenml/integrations/azure/orchestrators/__init__.py,sha256=q4rBPIJHcuUr6dLUBdrTkQGHFgPAqf44KE_I7mmOr0A,777
-zenml/integrations/azure/orchestrators/azureml_orchestrator.py,sha256=B3eUdXOKM1RdLRqPWiIF5cXtYZ9bM8bsbvOqCw7TbGw,20535
+zenml/integrations/azure/orchestrators/azureml_orchestrator.py,sha256=AXE_lm8pX_LrTB-mzhyVqKLV9w59V95k6MvKRiJNG6Y,20642
 zenml/integrations/azure/orchestrators/azureml_orchestrator_entrypoint_config.py,sha256=lacOyorsHa-HuD_kxN9M6BUiZDvs5jL9AnJWwrFCtp4,3104
 zenml/integrations/azure/service_connectors/__init__.py,sha256=yMz6bTCtIZqZwfEM6h7-PSWsd_DB8l0OD9z_bdzomZw,793
 zenml/integrations/azure/service_connectors/azure_service_connector.py,sha256=4Qma8FyKpZ5GujIa-RrGU2Lh80j9GHEyBwwDfyK7aAI,81344
 zenml/integrations/azure/step_operators/__init__.py,sha256=fV4_nAO0cH53x6_-F8-CbDEZwb_Vv64oq1r0-vtigEU,819
-zenml/integrations/azure/step_operators/azureml_step_operator.py,sha256=cmDLcxVd4lw_C2hz-_z50i9tcGgGbNtrBeL5FBQ3HTE,7394
+zenml/integrations/azure/step_operators/azureml_step_operator.py,sha256=bi6cl1PimxZF6D7PVZepC6OqHdGHTcsNGBe0njuiBCM,7397
 zenml/integrations/bentoml/__init__.py,sha256=vOlziqJ038cS7_ZtbGojWpefdQDigq8mLDALVC_lgbY,1835
 zenml/integrations/bentoml/constants.py,sha256=vZXwQbLgF4Vi2F_DICZlNnDcUa5FMejLZkjc0U1RpYo,788
 zenml/integrations/bentoml/flavors/__init__.py,sha256=7hzxI2zleFI9cplVJXn63iwL3s4XHeH1iup1Xc_dvIY,877
@@ -269,11 +269,11 @@ zenml/integrations/gcp/google_credentials_mixin.py,sha256=bPy3JYCCcyuTmPiVFqbY81
 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=sKAsi6F-Ik7XFgKVpdA_aNp3XUTIHKGAnBNIrBjrXNg,40794
+zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=IJwDUJsFrhhG9R49gi2yiWN48FfbX42EXTzgyOSm004,40873
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
 zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=9u-vEHbmSyN5IGwYI8v39TcFZg5ObgkxlbwSPz-e5zE,95018
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
-zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=X8CCniyAo7NHiy3Mv_YSKQ4Hw3UYMXob6B3uWKsCJ-0,13567
+zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=lV7IKfH02q8362ISeJxjir397Akv2xoEy22hvCyjayI,13570
 zenml/integrations/gcp/vertex_custom_job_parameters.py,sha256=B5RLkw7KDOi4ZfWHFnC6TGLphXMzToMjROxszCEAS9c,3676
 zenml/integrations/github/__init__.py,sha256=movoEHoX8SpY0t6mBS5D64enxG6lPkpYJIEQ9jElHLA,1427
 zenml/integrations/github/code_repositories/__init__.py,sha256=ub_hSE2ks2mZB1aeHRjQYz7QIRQIgOw2s080IIqJaGs,817
@@ -334,20 +334,20 @@ zenml/integrations/kubeflow/orchestrators/kubeflow_orchestrator.py,sha256=lPQSQn
 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=8feX6GpbefK0q3hOEVeUwP7-xgmWWmO-lSOSXmDUttw,10420
+zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=LDqxHiRtsbRdzskem9nkWC9wMbcg1ibNzG29_QydLB4,10806
 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=N66GH5ac22Xm_A3nr162kbFBhMeypSFaQjOQRHlGXIQ,18942
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=iuUllVHlXTvTftGk0Gs3EIITG0XXKwCT68UOfLHoJMs,27603
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=LTqbww_bQFnRS18VAPrxB356UoX8YJI6nlDR74tzqoo,13572
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=Om5fJ0U23TosOpr3QVVqrJP8KECpiVJcW_70ZndQkYQ,31657
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=74364kxOpI2hlEnnbnqMVq0UaaQnj_ozfkmB1Wst-gw,14815
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=QOwQnWCfB-t_BQ2eOZN0SakurGUd0GTMCSdUlREhk6I,2324
-zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=8_OU6cQxLRzkd1r40YNRoAnz5qHSFQOvVxHUhh88T04,13670
+zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=nw89WCsVDn9W-fmhKRjzzW1iOwlVn434IbcIsLPsfjw,14170
 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
 zenml/integrations/kubernetes/step_operators/__init__.py,sha256=40utDPYAezxHsFgO0UUIT_6XpCDzDapje6OH951XsTs,806
-zenml/integrations/kubernetes/step_operators/kubernetes_step_operator.py,sha256=lpkrA_barndyNLKqEUTaOIYJ6V8kHX9SfK5XFKK1md4,8848
+zenml/integrations/kubernetes/step_operators/kubernetes_step_operator.py,sha256=__7t6XuzedUEheUImSkX2grsDCz0YFWUJ6uxv47MXUo,8851
 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
@@ -400,7 +400,7 @@ zenml/integrations/modal/__init__.py,sha256=Cqc10MZP_dAFTk1QwvPCCyNouLM5OuewjpZ-
 zenml/integrations/modal/flavors/__init__.py,sha256=169Oirq-NUVl-2oiByrMcL3HBM0R971BrS6xVBk0DB4,922
 zenml/integrations/modal/flavors/modal_step_operator_flavor.py,sha256=r7xkzER0hvNY6N0CEZMiF1vRk7A7CfkuYOeGDJ517N4,3978
 zenml/integrations/modal/step_operators/__init__.py,sha256=8NYCXe7aNPnldCI9IjcHzJC_B5e4gtfGzpHNZ-ToKnE,780
-zenml/integrations/modal/step_operators/modal_step_operator.py,sha256=J2HiNaGnlP6kDLBnQKz8mQKaoaQLitFkjHhOP_XRWgI,8543
+zenml/integrations/modal/step_operators/modal_step_operator.py,sha256=ZndfvRsdNxdg-uW9ScezJUiAFo9Fd0ZKQd7pibrMUIM,8546
 zenml/integrations/neptune/__init__.py,sha256=PcqYVwybBcEy5YFODNA6bxT91kxaETReFmCWzExGhM8,1627
 zenml/integrations/neptune/experiment_trackers/__init__.py,sha256=DJi7lk7NXYrRg3VGPPSEsMycKECDfXL-h2V0A0r7z8Y,833
 zenml/integrations/neptune/experiment_trackers/neptune_experiment_tracker.py,sha256=hqEOgyBEQhjcnxvgw2oC2WZ0FAv1ez4VAqa_fJAHeCw,3716
@@ -525,7 +525,7 @@ zenml/integrations/spark/materializers/__init__.py,sha256=dZEkou5dpqq7bMcU0jgEcH
 zenml/integrations/spark/materializers/spark_dataframe_materializer.py,sha256=UXpwwz_8MLTVqfrN71Vsuck_GqDxl9D7YkKsY2K_4KA,2344
 zenml/integrations/spark/materializers/spark_model_materializer.py,sha256=bK6IUg1jRKVCF1-yUBoPtCtWEeIuaeX-cJO6pYz9WmE,2020
 zenml/integrations/spark/step_operators/__init__.py,sha256=NNgZXOk1S9yjOtQgYMrUkD9RUuV1-IkNvr6obfF0z8s,799
-zenml/integrations/spark/step_operators/kubernetes_step_operator.py,sha256=4GZfLP1bA1U7MCTHMC714X9LM_s_DQyxWSsmHx3-t0U,6295
+zenml/integrations/spark/step_operators/kubernetes_step_operator.py,sha256=R23cY6yqMAsS5srW7vLsVgawr2DkjuFzwgE9FbEuitY,6298
 zenml/integrations/spark/step_operators/spark_entrypoint_configuration.py,sha256=-zKDcOCMJ0mwDMi6MiTFhxkSqWfo8mLZ6cTqsLrMkwA,1336
 zenml/integrations/spark/step_operators/spark_step_operator.py,sha256=ZFvArHOS7PopwC96noXjz-rUgLyOX6y3UYtoBjJfia0,11501
 zenml/integrations/tekton/__init__.py,sha256=fR4Swt09E3VWCV-5to3zbCOaZcsDPtDs22zQrFWByhI,1656
@@ -649,7 +649,7 @@ zenml/models/v2/core/model_version_pipeline_run.py,sha256=JbPZZEQvOK9I32htkWdAON
 zenml/models/v2/core/pipeline.py,sha256=OXArih3YodMAZBS_GzuLL6VPpFv3c59EthwDfDZiNGk,12044
 zenml/models/v2/core/pipeline_build.py,sha256=z0LCc8aR8AdNFLLtzaAP0U0ffv7WpmSx9nNAyI__e14,17008
 zenml/models/v2/core/pipeline_deployment.py,sha256=nrJHrweNlwFsIdExINMF7jwhjOwBpOd8iwtEyAnvlz4,11971
-zenml/models/v2/core/pipeline_run.py,sha256=hWnM6Tf72uE9nxuSV6xacCMogOozuExvdtILgO6il98,32522
+zenml/models/v2/core/pipeline_run.py,sha256=h8kTuCplMec_IXi0IvP43ibx4JcRpBV9Hehaq2Yz1IQ,32578
 zenml/models/v2/core/project.py,sha256=fNNO8Tg5OhSzmFf2t6g4SpUzGWC96oHhUccVyWytvIE,5627
 zenml/models/v2/core/run_metadata.py,sha256=hRGQa_sk99uDSab3EyyOQhefypVpiQDCH3oAtblREDk,2432
 zenml/models/v2/core/run_template.py,sha256=6jdH1son7kpvFv-vtaOL2nXMATtVpqk_7a2xOVv_7cc,14097
@@ -682,18 +682,18 @@ zenml/models/v2/misc/statistics.py,sha256=ajce9rHC2bBylLzOmLfcOBSbTnJD67Y8n5YKCY
 zenml/models/v2/misc/tag.py,sha256=jUoz7wqMpDFlIUmvj4PVq8NYJJB7TMCcdRfW-DAABCU,974
 zenml/models/v2/misc/user_auth.py,sha256=1-yafNA9qK4wL8ToROjaklTVI7Mj9va0t80_4wm7w3U,6988
 zenml/orchestrators/__init__.py,sha256=Nhmc6U-q6c8TEH1Jb5l8XaKnc4KmLNspDpvvV5TcvzQ,2007
-zenml/orchestrators/base_orchestrator.py,sha256=1L2oJTNpVWUu5ndqyQPdcf2S5ebM7r2IHHrFatV6kPQ,14612
+zenml/orchestrators/base_orchestrator.py,sha256=xPakQ-_ft-Po_HYUQ7GL-VWbdVP0Zn4l65S4AzhT894,17508
 zenml/orchestrators/cache_utils.py,sha256=QkmTs-ANfXve9_QzTqgGlyulZDEWOngoTcsiSjG5aA8,5906
-zenml/orchestrators/containerized_orchestrator.py,sha256=rdebgBW0Bk--JcHcT0NpLkAbyhY0VS5xO1uwWEgkLpA,3230
-zenml/orchestrators/dag_runner.py,sha256=Ol5P24i8Oa0mmZeNZnUdd_HrQDawx8kP1epWa-b6f3g,9801
+zenml/orchestrators/containerized_orchestrator.py,sha256=-umxlNlQfezks1GLoi0SyXtXXo9dkC4Re9nA0fh_avw,3858
+zenml/orchestrators/dag_runner.py,sha256=Dz_Rn6Kew6mR5Upib2y5tYBjnCeFvHRaDv2l5d4iBSM,10675
 zenml/orchestrators/input_utils.py,sha256=IwDYEwhFVcvZcdi9Myg__UB1bkZfpa8xZ0XRSwc4-go,6934
 zenml/orchestrators/local/__init__.py,sha256=qlU91hgqGKeycw-Qntrn-iMuoMTaNrq-RgfOFeqwlM4,662
 zenml/orchestrators/local/local_orchestrator.py,sha256=KCzc901_wrb1DPTDu_IY6HFxTghe2oiLYPAdxEpj7K4,6033
 zenml/orchestrators/local_docker/__init__.py,sha256=k8J68ydy6HmmvE9tWo32g761H8P_Dw4AxWNf4UMpsbs,669
-zenml/orchestrators/local_docker/local_docker_orchestrator.py,sha256=RA88Yq8K9-zj4JkcVoA1dDDZZeA-rSFv5tUIW1tUYa8,9771
+zenml/orchestrators/local_docker/local_docker_orchestrator.py,sha256=dEc1R2_xzFIzvSwPirwtH2noYzPyw0gbdDhpsqAgFMM,10032
 zenml/orchestrators/output_utils.py,sha256=01vqke1ZfmfuLpgxNerF-QL2wA0VPv1zUdvlMw0OwUY,3508
-zenml/orchestrators/publish_utils.py,sha256=CSQKhx2f9r2knldDCuPR0lmVyRwI-Ps6Xbihfhxv21U,5477
-zenml/orchestrators/step_launcher.py,sha256=Dgimw9B2ZsrRPddZb6L25fmozUlnD8uBFykhPdS-SMM,17865
+zenml/orchestrators/publish_utils.py,sha256=OZ7tpYGfpNTe8ZY7PTHuZgSXIFtOhY8C7aRaXV3DFAk,8040
+zenml/orchestrators/step_launcher.py,sha256=Pi8EvxztomJlxzJWN3WtYv_lchhN4Ze59qi60tE0t_A,21498
 zenml/orchestrators/step_run_utils.py,sha256=ppCEAqB0UBtLCm4FuWjgeAS0wxAa8uivxgTOL7Abrn8,16228
 zenml/orchestrators/step_runner.py,sha256=EUgKG_g0fOQ6gnB1hPSSa6UXwUKVkguC-Yj-Q0yEQXg,26632
 zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7cM,5784
@@ -738,7 +738,7 @@ zenml/stack/__init__.py,sha256=vfHzaoRhPtS-XSlM8Vx1noJZDHF1Pj6LDz2knpn_QBg,1236
 zenml/stack/authentication_mixin.py,sha256=_Rn6SurHnyBuU_CJBOkwPR305yQFfqN60t6nEWJ4oa8,3991
 zenml/stack/flavor.py,sha256=wFLjajCw_G2NMR2UJzEGH2dcLj1dvmU9dDI5iS8b9rk,10544
 zenml/stack/flavor_registry.py,sha256=IL0fRrxxQJ9YkCYCeADP7nwWEQo4XBElJY4owMjKGbQ,6108
-zenml/stack/stack.py,sha256=3mLGNBuep0PeBcoFfOEaB6Ya206yav1ppiH5YEjp9xA,33042
+zenml/stack/stack.py,sha256=M2zZqh5RHTnQaHQCiIr5m_o7UkOLoS_AdXIk_mZItSo,33048
 zenml/stack/stack_component.py,sha256=aXCVf4txBW-jprUb8XRJ2r6D-OUrWg3QHkTtlIiIcd8,29255
 zenml/stack/stack_validator.py,sha256=hWbvvGIeWLj6NwSsF4GCc6RAxAWvxHXTcBZL9nJvcak,3111
 zenml/stack/utils.py,sha256=CHs9rxdqHkUT12KPhJX1YPtIWnZBoVlRlq5PzNymq3E,6406
@@ -790,6 +790,7 @@ zenml/utils/pipeline_docker_image_builder.py,sha256=Gp_eKg_LEdyBUSuZDC_PReyPHlqo
 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
+zenml/utils/run_utils.py,sha256=Zht6bM_Knnu0G-91XLUFeBmthBWkZ6lL3l7Ki32u918,2610
 zenml/utils/secret_utils.py,sha256=QmMVeCAt19-g48S8li8pHgqECK0ZndyO9Gne5rxvWsg,5901
 zenml/utils/server_utils.py,sha256=RzXeBcA0a-YdFX4BP9jSmLQ383uIj8SxbnXWBRjcfG4,1740
 zenml/utils/settings_utils.py,sha256=lAK13CiDCDkcLygizDbWB9q-9ukteVBJPypzFCrne9k,4631
@@ -1056,7 +1057,7 @@ zenml/zen_server/routers/plugin_endpoints.py,sha256=vyKKQIiLSzC5WkZoQdzS7sU-qYFS
 zenml/zen_server/routers/projects_endpoints.py,sha256=uL7oK8_0-H2nhjvPYHfYPvJOU4id7-n81MwjiSzHToE,8555
 zenml/zen_server/routers/run_metadata_endpoints.py,sha256=hUatO7c6LUBHpMDZYgDqiUbVUSBpt6-ZxaSC9eAaF9I,3688
 zenml/zen_server/routers/run_templates_endpoints.py,sha256=zUO8bTDQj7fwtQbnxPvkcdWHuhljpIK3YRMzCyp-jZc,8614
-zenml/zen_server/routers/runs_endpoints.py,sha256=NzIT6lRwy_mFr5EMc4f31W73Tu7JDyMc1l4YWcqiWng,13931
+zenml/zen_server/routers/runs_endpoints.py,sha256=qdwjCx0FMrFk0sLH5mBkvwE28XCJpCp4u4YVJqqzFEg,14039
 zenml/zen_server/routers/schedule_endpoints.py,sha256=m8TVhncapdtQqPm5EEm7CAeGr_i_0Tqa61EA8p1bIl4,5860
 zenml/zen_server/routers/secrets_endpoints.py,sha256=VgepInQHZXt-d8IAinftHs1n9il6hTD1qca6w5Pnnsk,9302
 zenml/zen_server/routers/server_endpoints.py,sha256=Aci-7TB5VZylmasEYc2tVpG7Npgtp6Dj56nrmy7mMmc,8133
@@ -1334,11 +1335,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=DrXGMkBxQIy2n_kkux5Xh2OM3Ks3MOpqP1D4aY8bfyY,7047
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
-zenml/zen_stores/sql_zen_store.py,sha256=Fj4AGUiEV-C1X-sXig-jeZEHCwih6Zr1BtoySOewF0o,482335
+zenml/zen_stores/sql_zen_store.py,sha256=Yosxc5JTU_XfAgLShgg_EZ1dN5AIL5txzU_fvXBHlYc,483118
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=weiSULdI9AsbCE10a5TcwtybX-BJs9hKhjPJnTapWv4,93023
-zenml_nightly-0.83.1.dev20250702.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.83.1.dev20250702.dist-info/METADATA,sha256=b6EA5qcrF0LsC_0WdWO-jmGbsnaCc4MtOpAC_zxhaxw,24316
-zenml_nightly-0.83.1.dev20250702.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.83.1.dev20250702.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.83.1.dev20250702.dist-info/RECORD,,
+zenml_nightly-0.83.1.dev20250704.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.83.1.dev20250704.dist-info/METADATA,sha256=p3rkQnm7Z0BQfEl51-eUVRpP4pztCkKt10YrFUz3OsQ,24316
+zenml_nightly-0.83.1.dev20250704.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.83.1.dev20250704.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.83.1.dev20250704.dist-info/RECORD,,