zenml-nightly 0.58.2.dev20240617__py3-none-any.whl → 0.58.2.dev20240618__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.58.2.dev20240617
+0.58.2.dev20240618

zenml/config/step_run_info.py

@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 """Step run info."""
 
+from typing import Any, Callable
 from uuid import UUID
 
 from zenml.config.pipeline_configurations import PipelineConfiguration
@@ -31,6 +32,8 @@ class StepRunInfo(StrictBaseModel):
     config: StepConfiguration
     pipeline: PipelineConfiguration
 
+    force_write_logs: Callable[..., Any]
+
     def get_image(self, key: str) -> str:
         """Gets the Docker image for the given key.
 

zenml/integrations/airflow/__init__.py

@@ -32,7 +32,11 @@ class AirflowIntegration(Integration):
     NAME = AIRFLOW
     # remove pendulum version requirement once Airflow supports
     # pendulum>-3.0.0
-    REQUIREMENTS = ["apache-airflow~=2.4.0", "pendulum<3.0.0"]
+    REQUIREMENTS = [
+        "apache-airflow~=2.4.0",
+        "pendulum<3.0.0",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]
 
     @classmethod
     def flavors(cls) -> List[Type[Flavor]]:

zenml/integrations/aws/step_operators/sagemaker_step_operator.py

@@ -261,7 +261,7 @@ class SagemakerStepOperator(BaseStepOperator):
                 "ExperimentName": settings.experiment_name,
                 "TrialName": sanitized_training_job_name,
             }
-
+        info.force_write_logs()
         estimator.fit(
             wait=True,
             inputs=inputs,

zenml/integrations/azure/step_operators/azureml_step_operator.py

@@ -297,4 +297,5 @@ class AzureMLStepOperator(BaseStepOperator):
         run = experiment.submit(config=run_config)
 
         run.display_name = info.run_name
+        info.force_write_logs()
         run.wait_for_completion(show_output=True)

zenml/integrations/constants.py

@@ -30,7 +30,6 @@ FEAST = "feast"
 GCP = "gcp"
 GITHUB = "github"
 GITLAB = "gitlab"
-GRAPHVIZ = "graphviz"
 HUGGINGFACE = "huggingface"
 HYPERAI = "hyperai"
 GREAT_EXPECTATIONS = "great_expectations"

zenml/integrations/deepchecks/__init__.py

@@ -41,6 +41,7 @@ class DeepchecksIntegration(Integration):
         "pandas<2.0.0",
         "opencv-python==4.5.5.64",  # pin to same version
         "opencv-python-headless==4.5.5.64",  # pin to same version
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
     ]
     APT_PACKAGES = ["ffmpeg", "libsm6", "libxext6"]
 

zenml/integrations/evidently/__init__.py

@@ -54,7 +54,10 @@ class EvidentlyIntegration(Integration):
     """[Evidently](https://github.com/evidentlyai/evidently) integration for ZenML."""
 
     NAME = EVIDENTLY
-    REQUIREMENTS = ["evidently>0.2.6,<0.4.5"]  # supports pyyaml 6
+    REQUIREMENTS = [
+        "evidently>0.2.6,<0.4.5",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]  # supports pyyaml 6
 
     @classmethod
     def flavors(cls) -> List[Type[Flavor]]:

zenml/integrations/gcp/step_operators/vertex_step_operator.py

@@ -274,6 +274,7 @@ class VertexStepOperator(BaseStepOperator, GoogleCredentialsMixin):
             custom_job["display_name"],
             parent,
         )
+        info.force_write_logs()
         response = client.create_custom_job(
             parent=parent, custom_job=custom_job
         )

zenml/integrations/langchain/__init__.py

@@ -25,7 +25,11 @@ class LangchainIntegration(Integration):
     """Definition of langchain integration for ZenML."""
 
     NAME = LANGCHAIN
-    REQUIREMENTS = ["langchain==0.0.325", "pyyaml>=6.0.1"]
+    REQUIREMENTS = [
+        "langchain==0.0.325",
+        "pyyaml>=6.0.1",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]
 
     @classmethod
     def activate(cls) -> None:

zenml/integrations/neural_prophet/__init__.py

@@ -21,7 +21,11 @@ class NeuralProphetIntegration(Integration):
     """Definition of NeuralProphet integration for ZenML."""
 
     NAME = NEURAL_PROPHET
-    REQUIREMENTS = ["neuralprophet>=0.3.2,<0.5.0", "holidays>=0.4.1,<0.25.0"]
+    REQUIREMENTS = [
+        "neuralprophet>=0.3.2,<0.5.0",
+        "holidays>=0.4.1,<0.25.0",
+        "tenacity!=8.4.0",  # https://github.com/jd/tenacity/issues/471
+    ]
 
     @classmethod
     def activate(cls) -> None:

zenml/integrations/spark/step_operators/spark_step_operator.py

@@ -303,6 +303,8 @@ class SparkStepOperator(BaseStepOperator):
         # Add any additional configuration given by the user.
         self._additional_configuration(spark_config=conf, settings=settings)
 
+        info.force_write_logs()
+
         # Generate a spark-submit command given the configuration
         self._launch_spark_job(
             spark_config=conf,

zenml/logging/step_logging.py

@@ -98,7 +98,7 @@ def prepare_logs_uri(
         artifact_store.makedirs(logs_uri_folder)
         return logs_uri_folder
     else:
-        logs_uri = os.path.join(logs_base_uri, f"{log_key}.log")
+        logs_uri = os.path.join(logs_base_uri, f"{log_key}{LOGS_EXTENSION}")
         if artifact_store.exists(logs_uri):
             logger.warning(
                 f"Logs file {logs_uri} already exists! Removing old log file..."
@@ -238,7 +238,7 @@ class StepLogsStorage:
         # Immutable filesystems state
         self.last_merge_time = time.time()
         self.log_files_not_merged: List[str] = []
-        self.next_merged_file_name: str = f"{time.time()}{LOGS_EXTENSION}"
+        self.next_merged_file_name: str = self._get_timestamped_filename()
 
     @property
     def artifact_store(self) -> "BaseArtifactStore":
@@ -276,6 +276,14 @@ class StepLogsStorage:
             or time.time() - self.last_save_time >= self.time_interval
         )
 
+    def _get_timestamped_filename(self) -> str:
+        """Returns a timestamped filename.
+
+        Returns:
+            The timestamped filename.
+        """
+        return f"{time.time()}{LOGS_EXTENSION}"
+
     def save_to_file(self, force: bool = False) -> None:
         """Method to save the buffer to the given URI.
 
@@ -293,14 +301,14 @@ class StepLogsStorage:
                     if self.artifact_store.config.IS_IMMUTABLE_FILESYSTEM:
                         if not self.log_files_not_merged:
                             self.next_merged_file_name = (
-                                f"{time.time()}{LOGS_EXTENSION}"
+                                self._get_timestamped_filename()
                             )
-                        log_file_ = f"{time.time()}{LOGS_EXTENSION}"
-                        self.log_files_not_merged.append(log_file_)
+                        _logs_uri = self._get_timestamped_filename()
+                        self.log_files_not_merged.append(_logs_uri)
                         with self.artifact_store.open(
                             os.path.join(
                                 self.logs_uri,
-                                log_file_,
+                                _logs_uri,
                             ),
                             "w",
                         ) as file:
@@ -360,7 +368,7 @@ class StepLogsStorage:
         """
         if self.artifact_store.config.IS_IMMUTABLE_FILESYSTEM:
             files_ = files or self.artifact_store.listdir(self.logs_uri)
-            file_name_ = file_name or f"full_log{LOGS_EXTENSION}"
+            file_name_ = file_name or self._get_timestamped_filename()
             if len(files_) > 1:
                 files_.sort()
                 logger.debug("Log files count: %s", len(files_))

zenml/orchestrators/step_launcher.py

@@ -17,7 +17,8 @@ import os
 import time
 from contextlib import nullcontext
 from datetime import datetime
-from typing import TYPE_CHECKING, Dict, Optional, Tuple
+from functools import partial
+from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple
 
 from zenml.client import Client
 from zenml.config.step_configurations import Step
@@ -247,10 +248,28 @@ class StepLauncher:
                     while retries < max_retries:
                         last_retry = retries == max_retries - 1
                         try:
+                            # here pass a forced save_to_file callable to be
+                            # used as a dump function to use before starting
+                            # the external jobs in step operators
+                            if isinstance(
+                                logs_context,
+                                step_logging.StepLogsStorageContext,
+                            ):
+                                force_write_logs = partial(
+                                    logs_context.storage.save_to_file,
+                                    force=True,
+                                )
+                            else:
+
+                                def _bypass() -> None:
+                                    return None
+
+                                force_write_logs = _bypass
                             self._run_step(
                                 pipeline_run=pipeline_run,
                                 step_run=step_run_response,
                                 last_retry=last_retry,
+                                force_write_logs=force_write_logs,
                             )
                             logger.info(
                                 f"Step `{self._step_name}` completed successfully."
@@ -324,9 +343,11 @@ class StepLauncher:
             user=client.active_user.id,
             workspace=client.active_workspace.id,
             deployment=self._deployment.id,
-            pipeline=self._deployment.pipeline.id
-            if self._deployment.pipeline
-            else None,
+            pipeline=(
+                self._deployment.pipeline.id
+                if self._deployment.pipeline
+                else None
+            ),
             status=ExecutionStatus.RUNNING,
             orchestrator_environment=get_run_environment_dict(),
             start_time=datetime.utcnow(),
@@ -415,6 +436,7 @@ class StepLauncher:
         self,
         pipeline_run: PipelineRunResponse,
         step_run: StepRunResponse,
+        force_write_logs: Callable[..., Any],
         last_retry: bool = True,
     ) -> None:
         """Runs the current step.
@@ -422,6 +444,7 @@ class StepLauncher:
         Args:
             pipeline_run: The model of the current pipeline run.
             step_run: The model of the current step run.
+            force_write_logs: The context for the step logs.
             last_retry: Whether this is the last retry of the step.
         """
         # Prepare step run information.
@@ -432,6 +455,7 @@ class StepLauncher:
             pipeline_step_name=self._step_name,
             run_id=pipeline_run.id,
             step_run_id=step_run.id,
+            force_write_logs=force_write_logs,
         )
 
         output_artifact_uris = output_utils.prepare_output_artifact_uris(

zenml/step_operators/step_operator_entrypoint_configuration.py

@@ -86,6 +86,7 @@ class StepOperatorEntrypointConfiguration(StepEntrypointConfiguration):
             pipeline_step_name=self.entrypoint_args[STEP_NAME_OPTION],
             run_id=pipeline_run.id,
             step_run_id=step_run_id,
+            force_write_logs=lambda: None,
         )
 
         stack = Client().active_stack

zenml/utils/pipeline_docker_image_builder.py

@@ -119,6 +119,7 @@ class PipelineDockerImageBuilder:
             ValueError: If no Dockerfile and/or custom parent image is
                 specified and the Docker configuration doesn't require an
                 image build.
+            ValueError: If the specified Dockerfile does not exist.
         """
         requirements: Optional[str] = None
         dockerfile: Optional[str] = None
@@ -133,6 +134,14 @@ class PipelineDockerImageBuilder:
             # pipeline?
             return docker_settings.parent_image, dockerfile, requirements
 
+        if docker_settings.dockerfile and not os.path.isfile(
+            docker_settings.dockerfile
+        ):
+            raise ValueError(
+                "Dockerfile at path "
+                f"{os.path.abspath(docker_settings.dockerfile)} not found."
+            )
+
         stack.validate()
         image_builder = stack.image_builder
         if not image_builder:

{zenml_nightly-0.58.2.dev20240617.dist-info → zenml_nightly-0.58.2.dev20240618.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenml-nightly
-Version: 0.58.2.dev20240617
+Version: 0.58.2.dev20240618
 Summary: ZenML: Write production-ready ML code.
 Home-page: https://zenml.io
 License: Apache-2.0
@@ -81,6 +81,7 @@ Requires-Dist: mkdocs-material (>=8.1.7,<9.0.0) ; extra == "dev"
 Requires-Dist: mkdocstrings (>=0.17.0,<0.18.0) ; extra == "dev"
 Requires-Dist: mlstacks (==0.8.0) ; extra == "mlstacks"
 Requires-Dist: mypy (==1.7.1) ; extra == "dev"
+Requires-Dist: numpy (<2.0.0)
 Requires-Dist: orjson (>=3.10.0,<3.11.0) ; extra == "server"
 Requires-Dist: pandas (>=1.1.5)
 Requires-Dist: passlib[bcrypt] (>=1.7.4,<1.8.0)

{zenml_nightly-0.58.2.dev20240617.dist-info → zenml_nightly-0.58.2.dev20240618.dist-info}/RECORD

@@ -6,7 +6,7 @@ RELEASE_NOTES.md,sha256=wu9pumMNKuHhAViSuGVGlhuC_PakKaAois07lk0tk0M,330862
 ROADMAP.md,sha256=hiLSmr16BH8Dfx7SaQM4JcXCGCVl6mFZPFAwJeDTrJU,407
 SECURITY.md,sha256=9DepA8y03yvCZLHEfcXLTDH4lUyKHquAdukBsccNN7c,682
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=kvEUTmIWNcxIvYGw8QwWCi5Lw9qnJmoM9WcUd1MtKIA,19
+zenml/VERSION,sha256=Nr0I7_xlEyXmKT3MVoC0Hac3V3igd-l3h5N_8NTCnMc,19
 zenml/__init__.py,sha256=qT3QHiuTgUwk05sOiML0Gcjre3pugvDAw84_br9Psvo,2394
 zenml/_hub/__init__.py,sha256=6qDzpQAAZa__Aiiz0mC1qM-9dw9_jk_v_aXeJknxbDE,644
 zenml/_hub/client.py,sha256=LAxLuDjY9ya1PPRzxS0_pvu__FSpAqEjbtHTqVwgMWc,9071
@@ -94,7 +94,7 @@ zenml/config/server_config.py,sha256=2GZM6HuExXZcBKmfY-HzgRyiUCTAKCUL0QvUG_FMA3w
 zenml/config/settings_resolver.py,sha256=vb7cWaPFw1KsHv33EMBObmC-_R7lmJFeSLC5l4u9RPU,4279
 zenml/config/source.py,sha256=jFqmzjURZ1Op0QupoKDIfw7bcak3uuASI-sh1NfB54Y,6894
 zenml/config/step_configurations.py,sha256=AdDCB54-FYc1iQojo0HppzDh-1R1Mzz0nU2iVXtut_k,8792
-zenml/config/step_run_info.py,sha256=bZW28YZ1M9sG4EAKHthCAKm6DsSR9ox1OkJG0Zjfzhg,1813
+zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I,1888
 zenml/config/store_config.py,sha256=xdxwKt1T4JmO_U0_2_mptLPb41FEpT7dLz-liJAeu80,3585
 zenml/config/strict_base_model.py,sha256=jB7a9MKKjv01I90PIzMTfnF-VWIHuO1xg5HMTmh8JFc,912
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
@@ -137,7 +137,7 @@ zenml/image_builders/build_context.py,sha256=7z73NWiHtv9pg2TFTzhcx-DJgyzGuljKEKP
 zenml/image_builders/local_image_builder.py,sha256=nxwzPGgB2ePE51HcvT6hM6w37j9gn2ITEJuPMrx_SKw,5709
 zenml/integrations/README.md,sha256=hFIZwjsAItHjvDWVBqGSF-ZAeMsFR2GKX1Axl2g1Bz0,6190
 zenml/integrations/__init__.py,sha256=6XUrOV9KjmDxegCkNfX0PIIQoowek0suYJJVF4X_LIk,4424
-zenml/integrations/airflow/__init__.py,sha256=03N2nhvFUpn5_aJStVjjVQYIedqZ5BXxH4uUA2MBdC4,1771
+zenml/integrations/airflow/__init__.py,sha256=hm_KBce12WDKvg8QqEC7WI9AbeSIrUypSHNYguX6jvI,1866
 zenml/integrations/airflow/flavors/__init__.py,sha256=Y48mn5OxERPPaXDBd5CFAIn6yhLPsgN5ZMk26hLXiNM,800
 zenml/integrations/airflow/flavors/airflow_orchestrator_flavor.py,sha256=M5hVqkPqZto3xz1xiU7Z4dNhY7aD88KixtbicHPwTu0,6071
 zenml/integrations/airflow/orchestrators/__init__.py,sha256=UPly5wqlFdzLSiCeEtIExRXjTNdO0ys2tHSEbEv1Dks,845
@@ -161,7 +161,7 @@ zenml/integrations/aws/orchestrators/sagemaker_orchestrator_entrypoint_config.py
 zenml/integrations/aws/service_connectors/__init__.py,sha256=w2Md40yG89PwmU9eBceh6dGy3XYZ3MKusNAZ51sGOgE,783
 zenml/integrations/aws/service_connectors/aws_service_connector.py,sha256=T77DHFZovaKh240SnP6i9PuHC1BSR_u9tND1WH6bhqY,91990
 zenml/integrations/aws/step_operators/__init__.py,sha256=HK5oIqLixYKUKoiN4LnqTyGjAZJUdUqWYGqJhFSWyo0,817
-zenml/integrations/aws/step_operators/sagemaker_step_operator.py,sha256=FU0C1M50GV1mJfmDruF9okPGgV23ozPhWHirsfvaRtI,10082
+zenml/integrations/aws/step_operators/sagemaker_step_operator.py,sha256=-y1zqfIDJm2wwy_KETJ1nHdKuSbIHPFK3bT-0i9tpoA,10113
 zenml/integrations/aws/step_operators/sagemaker_step_operator_entrypoint_config.py,sha256=2cXroe6-bXyHVkO5yPnZagNpqx6MrMDcvuvNr8J2j-A,1581
 zenml/integrations/azure/__init__.py,sha256=v147xzcv4pkbiqAxNYYhWUWN9rjQNYsbblMT0BhiT3M,2515
 zenml/integrations/azure/artifact_stores/__init__.py,sha256=dlIwbpgjE0Hy4rhMbelNJHVKm4t8tj_hRu9mQ_cEIAg,820
@@ -172,7 +172,7 @@ zenml/integrations/azure/flavors/azureml_step_operator_flavor.py,sha256=QR5gp1Wp
 zenml/integrations/azure/service_connectors/__init__.py,sha256=yMz6bTCtIZqZwfEM6h7-PSWsd_DB8l0OD9z_bdzomZw,793
 zenml/integrations/azure/service_connectors/azure_service_connector.py,sha256=AqBWodE3dpZhpK5xjINs_-uflKubG2pYbBxkPppOYHY,76373
 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=vGxTbQZqNsCbKFvNIrcY4Qa5SXlL4lgwcvXRh6tSIe0,10661
+zenml/integrations/azure/step_operators/azureml_step_operator.py,sha256=M_Uc1mecwz9OmqhfF73MCWZJr-yABf7taSUnwT4KWnw,10693
 zenml/integrations/bentoml/__init__.py,sha256=q7Vo8DL4j5N3p2Gu0_vio5VUUlW8wQ6N-ZcunK_ibiY,1876
 zenml/integrations/bentoml/constants.py,sha256=vZXwQbLgF4Vi2F_DICZlNnDcUa5FMejLZkjc0U1RpYo,788
 zenml/integrations/bentoml/flavors/__init__.py,sha256=7hzxI2zleFI9cplVJXn63iwL3s4XHeH1iup1Xc_dvIY,877
@@ -196,8 +196,8 @@ zenml/integrations/comet/experiment_trackers/__init__.py,sha256=reGygyAEgMrlc-9Q
 zenml/integrations/comet/experiment_trackers/comet_experiment_tracker.py,sha256=JnB_TqiCD8t9t6cVxWoomxvBuhA4jIJHYFZ-gKdGXf8,5767
 zenml/integrations/comet/flavors/__init__.py,sha256=x-XK-YwHMxz3zZPoIXo3X5vq_5VYUJAnsIoEX_ZooOU,883
 zenml/integrations/comet/flavors/comet_experiment_tracker_flavor.py,sha256=Sk8iEKOCy-icytdwg83MuVtwE-NYb6e4RA4FQ2dlOYw,3779
-zenml/integrations/constants.py,sha256=3dphkzUcQhQ8FTeU5DcxE_mxaqR1sxcjNQIVK9KgC6w,1935
-zenml/integrations/deepchecks/__init__.py,sha256=N8fMQuUXA2DzPhHvfUJEmGFJwB-2e4tnamEwLrzHEFU,2263
+zenml/integrations/constants.py,sha256=tF6joxaMMuEjpmHUV32_0tQrgk8gxL493Mqiys0vPoE,1913
+zenml/integrations/deepchecks/__init__.py,sha256=_3a8nMNzWQxWBPzFEZWzQVP4is1BNWYJYgLP7IHweZM,2335
 zenml/integrations/deepchecks/data_validators/__init__.py,sha256=i7QKjkqoUSFGg_l7JuVbnHFs5uxOKRcSp0s3apwF2RM,835
 zenml/integrations/deepchecks/data_validators/deepchecks_data_validator.py,sha256=7lBWLbYrKljbAMsJZeFfLnVtPNWc9rmqb8JaSKhURPM,19668
 zenml/integrations/deepchecks/flavors/__init__.py,sha256=TkUMrj_WWzGiPqc4GxCYjE-80AQyYhYDUBoqZ9fWTOc,819
@@ -219,7 +219,7 @@ zenml/integrations/discord/flavors/discord_alerter_flavor.py,sha256=9hX7R7dfxSwi
 zenml/integrations/discord/steps/__init__.py,sha256=stSDntUMzrHzwMJm1V1-jm7otII7uW6Fxj7qYB7MWrc,663
 zenml/integrations/discord/steps/discord_alerter_ask_step.py,sha256=puBERGjhpBRaift8GCygAgnjgZHbeqclRywxJjjjEG8,2553
 zenml/integrations/discord/steps/discord_alerter_post_step.py,sha256=te4M4Q47e1nShPHLLv414bjDuG_r7XCxDUbLgwGXEtI,2283
-zenml/integrations/evidently/__init__.py,sha256=PJOuxYP3h0JYmdkzdirTx6r9tyvdtXbbqMC3v8819bQ,2471
+zenml/integrations/evidently/__init__.py,sha256=lQDaCJH7HqZmLvWOkV2m5zy8Mga_O1oOEa_xqi50GY4,2558
 zenml/integrations/evidently/column_mapping.py,sha256=AVC-rDawczow3vgmhKXif7iaMBlIE_hYLAVZ2G7a8is,3350
 zenml/integrations/evidently/data_validators/__init__.py,sha256=7H1HCXAefk-asnSAYqfud-l17rsBFfhCrgps2abhmFY,830
 zenml/integrations/evidently/data_validators/evidently_data_validator.py,sha256=ES54WRMnDeRHvKTaRmstdXckmO-3TIggIBgogXX6NaM,9706
@@ -259,7 +259,7 @@ zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=Xhe8B050Bi5pB
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
 zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=7LYlSid_jjP_mCy8ZSXiXOEc2ThknXs2Xc4agGvVuhw,78040
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
-zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=tlcvrU4Zv70IcvN6Au_FOJ2DEtWys8kbLGu3jGrs0Z4,13460
+zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=qPQOSacCFUcXCl0MkuzVkxpewU8NAPjiGQsjBF1N0ro,13492
 zenml/integrations/github/__init__.py,sha256=A8Yd--BbAG3HEfbWYOIEy_kzyLs2tBiawiLMosXd1Do,1467
 zenml/integrations/github/code_repositories/__init__.py,sha256=ub_hSE2ks2mZB1aeHRjQYz7QIRQIgOw2s080IIqJaGs,817
 zenml/integrations/github/code_repositories/github_code_repository.py,sha256=G_cVWIiwYlnGAyTMpFoiHi8i3fZ9rVKrFjRxkoseIhc,6630
@@ -339,7 +339,7 @@ zenml/integrations/label_studio/label_config_generators/label_config_generators.
 zenml/integrations/label_studio/label_studio_utils.py,sha256=NelKDXCoEIF37-xh7rffHeuHEwWvkfshR5w5f6HuBII,3316
 zenml/integrations/label_studio/steps/__init__.py,sha256=SQ-6oyRtqHDsU-QjOdvtd-cD8plsW40Dwl5SZnWtbbA,895
 zenml/integrations/label_studio/steps/label_studio_standard_steps.py,sha256=k7UTFzDZBTdV0NbVtRKMqQo-gURvdSMtjtHoFfiIWgs,8695
-zenml/integrations/langchain/__init__.py,sha256=CgsMZtE6FG9EFt-_l8LqXuXThjWVX_oAIUe-PBZ0zXY,1225
+zenml/integrations/langchain/__init__.py,sha256=HIpUU_qqULjQA6-hLkhteg1sfo5dTk1czCyOhug7Szs,1320
 zenml/integrations/langchain/materializers/__init__.py,sha256=ouU6MDX_gZc0FVgNK8xO6F7B2XOEikrevQEZpdYyaOM,1037
 zenml/integrations/langchain/materializers/document_materializer.py,sha256=p_QxwiHuOsk6qGy_T5AZMoLGUGeh8M30mDFDhB3mrQU,1239
 zenml/integrations/langchain/materializers/openai_embedding_materializer.py,sha256=WLErmqNJU5gmc-EqMEsIOAdrpyZPB62th9r4Xx0g-Ao,1294
@@ -376,7 +376,7 @@ zenml/integrations/neptune/experiment_trackers/run_state.py,sha256=vY34dDSadyzuI
 zenml/integrations/neptune/flavors/__init__.py,sha256=NzORpmtI3Vu7yH1maj5pYd_2gQoN4QR2ZZLy1uPr6uw,975
 zenml/integrations/neptune/flavors/neptune_experiment_tracker_flavor.py,sha256=WkYmLCXAk8XzOvO34ueYoUX3BicIqrhcANjewWFkrVs,3622
 zenml/integrations/neptune/neptune_constants.py,sha256=-VddhrALS1HMBhGtFiGDKaRahC-qhG0JAF2zAuc3lMM,746
-zenml/integrations/neural_prophet/__init__.py,sha256=9qqveTAN0BO_QN3UpMNfohDm0V_Mf5fx3XoM44-rOuY,1208
+zenml/integrations/neural_prophet/__init__.py,sha256=nQ5aBqKkzAukrEc7_9E1nU1iRhTHqhw0yAtEIk7dhys,1303
 zenml/integrations/neural_prophet/materializers/__init__.py,sha256=7O-qwxLjxLmvqeUgeLWBOHe5pvxwDFRg5hCOhffyd0k,802
 zenml/integrations/neural_prophet/materializers/neural_prophet_materializer.py,sha256=U8OvRObumGAk66UJFHLKWtez-I3wU2zd1j-XvCbSBr4,1570
 zenml/integrations/openai/__init__.py,sha256=Zi9LaUe6sElglthZQ4iuqjO34v5DWZoltj387uPZTCs,966
@@ -482,7 +482,7 @@ zenml/integrations/spark/materializers/spark_model_materializer.py,sha256=bK6IUg
 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/spark_entrypoint_configuration.py,sha256=-zKDcOCMJ0mwDMi6MiTFhxkSqWfo8mLZ6cTqsLrMkwA,1336
-zenml/integrations/spark/step_operators/spark_step_operator.py,sha256=cOx26G6AeFnb2GmNvqWB7efoTY0ageSlHsBq02F5Bno,11468
+zenml/integrations/spark/step_operators/spark_step_operator.py,sha256=ZFvArHOS7PopwC96noXjz-rUgLyOX6y3UYtoBjJfia0,11501
 zenml/integrations/tekton/__init__.py,sha256=0wU-1dxHCy6hSvomqttpQPUVFptYT1G_ngaMc6AhBQk,1629
 zenml/integrations/tekton/flavors/__init__.py,sha256=-S5XuwcZKWyGb9tVfl_gEFJj1KSXX6kwIfe-EhE0TmM,864
 zenml/integrations/tekton/flavors/tekton_orchestrator_flavor.py,sha256=WpmIlM-0MfGbOf7Vg-yJVLHu8cvYz-me_QuhRBBGEoE,6494
@@ -533,7 +533,7 @@ zenml/lineage_graph/node/base_node.py,sha256=-eH0CcxXkrfF_F1NhV1jMFzbjHvolpjl_od
 zenml/lineage_graph/node/step_node.py,sha256=Tr9HbkBgcvpjKamX5cwPZM7i5LE66kU84_ATzVmKeMg,1300
 zenml/logger.py,sha256=5K5csgOLAecvUOYWWSaqirgkC1sZkZ4Ka7JALAGL_Ik,6131
 zenml/logging/__init__.py,sha256=lnqbOa31wAHwPP5f8vZazOrUwnP2QviLiIVwxoAefD8,975
-zenml/logging/step_logging.py,sha256=HJ1CKSJ9MYLbde7EyFUEhwpKReh_gF16MOcBraiCs10,16813
+zenml/logging/step_logging.py,sha256=WDqKcQnKEs3hmWOaY1knttWRdunmJ1_PgmqrK7GRtd4,17038
 zenml/materializers/__init__.py,sha256=YE4Eh-ftvnfKaqbwz4TylODPCGHBdMj2lJFjNtmjoko,1767
 zenml/materializers/base_materializer.py,sha256=zvuden4-YQ1SO_lgwuenmt54xuUyFg-dw0aKqXKDDKg,10291
 zenml/materializers/built_in_materializer.py,sha256=x8fCklOmBT9M83CkcmV8o4Qukxi7lEbarmZjIVNYJwk,14948
@@ -637,7 +637,7 @@ zenml/orchestrators/local_docker/__init__.py,sha256=k8J68ydy6HmmvE9tWo32g761H8P_
 zenml/orchestrators/local_docker/local_docker_orchestrator.py,sha256=Dxx3PjmEukrUxVajChmqj8tqBKNNDG46D8eefTM6g2w,9263
 zenml/orchestrators/output_utils.py,sha256=Gz7SX2cbQ3w4eyfU0XuhKEKGtalQGBoc6moDRNVHN8M,3311
 zenml/orchestrators/publish_utils.py,sha256=aNwgTDmVSq9qCDP3Ldk77YNXnWx_YHjYNTEJwYeZo9s,4579
-zenml/orchestrators/step_launcher.py,sha256=ub8AJnTqIzn7wMUMa6SvesQ9eKUEiyWkNzsTLRlNA9g,20115
+zenml/orchestrators/step_launcher.py,sha256=uEJ55LtE_oVT8jp6seiifRaj5wkBPAzG1pGgliD6KbM,21241
 zenml/orchestrators/step_runner.py,sha256=e_HXF5i1rJ06jddGL8TtBvNAPzOxxdR1VzorSIuoSro,26496
 zenml/orchestrators/topsort.py,sha256=D8evz3X47zwpXd90NMLsJD-_uCeXtV6ClzNfDUrq7cM,5784
 zenml/orchestrators/utils.py,sha256=osonxW0DZxDAHnHLKHcDRA63VMsZNjpg2BHY55lBVCg,10498
@@ -688,7 +688,7 @@ zenml/stack/stack_validator.py,sha256=hWbvvGIeWLj6NwSsF4GCc6RAxAWvxHXTcBZL9nJvca
 zenml/stack/utils.py,sha256=W_ZKttALLGABYSzqapkIdHerbOWf6gRL8Tl9Bgj_q5s,5460
 zenml/step_operators/__init__.py,sha256=tqj7fgnQyZubLjwUu4ITwkA-70KMQz4g37agbfF7ZgE,1228
 zenml/step_operators/base_step_operator.py,sha256=ZRnY6lcEHY8xZskrKKdPhgKg2BlEoh2_kb8xCaM2D1g,3522
-zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=-Tin2h4fXd-N4giOge9dq0J2WkcJdpqxyyw6zFF4oSA,3647
+zenml/step_operators/step_operator_entrypoint_configuration.py,sha256=vG0Z9D2C72PYOZ_P3eXYTmm2TvL-dS_kHhSMq-coAKE,3690
 zenml/steps/__init__.py,sha256=IvxepI3tYp6LiVHFJnet9xZ7yFlH6NHCAI69SvQsfaI,1682
 zenml/steps/base_parameters.py,sha256=rKnALd0lMv0jOX8E9TwHJkeP3H-ijEFXY0T5dMCu9EQ,756
 zenml/steps/base_step.py,sha256=2syYeLgmPwKPGKNp4mrXnSlZ3RVslXrvWGDFBXiK6EY,49665
@@ -720,7 +720,7 @@ zenml/utils/mlstacks_utils.py,sha256=Jv0ZRK8seBcnA96UelJhRaB6mRhZiA4062yE0yE7RaA
 zenml/utils/networking_utils.py,sha256=zTDbOMkMPRWiWLZ2ccchd37rvvZWIIh0Kr9dZE-VJAs,7845
 zenml/utils/package_utils.py,sha256=IrUYRkedfgJ6O6R-bmmpdrO_SS33QFsVJZOE61uM2vk,1677
 zenml/utils/pagination_utils.py,sha256=nY6jg_RCQoeCipHrQfkdEarCah88P17IHMb4kswUhXY,1341
-zenml/utils/pipeline_docker_image_builder.py,sha256=_gq0KvtwIcHTfb-ty5CyWJ8iIZKFT2puigSDntbGXw8,26879
+zenml/utils/pipeline_docker_image_builder.py,sha256=mkIZyPrfUPbSduMC5nePlVnHPcC6q-IkVdEd-oOBZT8,27218
 zenml/utils/proxy_utils.py,sha256=fgRlLa9pfXJDoxtB31_YP7DClOMQLek_nMmM0et6i3w,7241
 zenml/utils/pydantic_utils.py,sha256=pDpv_ciIj7o5RIgwWa0W6BUjmHjltcjWK4RRSoGVR2A,8468
 zenml/utils/secret_utils.py,sha256=0b0FeMhr28oxe4t6lOhnmHjENFADACynUEKeu6gBSkE,4019
@@ -1294,8 +1294,8 @@ zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=0CgWJo
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Cips_SPX7CtfYQyOZNiHAhmVtVgb17HCyHQia5MlTX4,8653
 zenml/zen_stores/sql_zen_store.py,sha256=f5spF_vWJg8kO6ud9uzELqh7LRxSAWyB1HLono1_gT0,353540
 zenml/zen_stores/zen_store_interface.py,sha256=vi0tr6vDES97h_rFZZYeaz2Y0nM4xm66CcUXSOEITN4,86372
-zenml_nightly-0.58.2.dev20240617.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.58.2.dev20240617.dist-info/METADATA,sha256=7PWfOtF1VF_n6npSaJN-8gNivr5K-UIT7ocAxXBGnz4,20021
-zenml_nightly-0.58.2.dev20240617.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zenml_nightly-0.58.2.dev20240617.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.58.2.dev20240617.dist-info/RECORD,,
+zenml_nightly-0.58.2.dev20240618.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.58.2.dev20240618.dist-info/METADATA,sha256=LyKsWSGuxRlx9vG5MS-VJXIdG9pVdQJJDXpxt8hymXo,20051
+zenml_nightly-0.58.2.dev20240618.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zenml_nightly-0.58.2.dev20240618.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.58.2.dev20240618.dist-info/RECORD,,