zenml-nightly 0.80.1.dev20250330__py3-none-any.whl → 0.80.1.dev20250401__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.80.1.dev20250330
+0.80.1.dev20250401

zenml/integrations/gcp/orchestrators/vertex_orchestrator.py

@@ -736,6 +736,9 @@ class VertexOrchestrator(ContainerizedOrchestrator, GoogleCredentialsMixin):
                         "Waiting for the Vertex AI Pipelines job to finish..."
                     )
                     run.wait()
+                    logger.info(
+                        "Vertex AI Pipelines job completed successfully."
+                    )
 
         except google_exceptions.ClientError as e:
             logger.error("Failed to create the Vertex AI Pipelines job: %s", e)

zenml/logging/step_logging.py

@@ -253,7 +253,12 @@ class StepLogsStorage:
             return
 
         if not self.disabled:
-            self.buffer.append(text)
+            # Add timestamp to the message when it's received
+            timestamp = utc_now().strftime("%Y-%m-%d %H:%M:%S")
+            formatted_message = (
+                f"[{timestamp} UTC] {remove_ansi_escape_codes(text)}"
+            )
+            self.buffer.append(formatted_message)
             self.save_to_file()
 
     @property
@@ -325,23 +330,13 @@ class StepLogsStorage:
                             "w",
                         ) as file:
                             for message in self.buffer:
-                                timestamp = utc_now().strftime(
-                                    "%Y-%m-%d %H:%M:%S"
-                                )
-                                file.write(
-                                    f"[{timestamp} UTC] {remove_ansi_escape_codes(message)}\n"
-                                )
+                                file.write(f"{message}\n")
                     else:
                         with self.artifact_store.open(
                             self.logs_uri, "a"
                         ) as file:
                             for message in self.buffer:
-                                timestamp = utc_now().strftime(
-                                    "%Y-%m-%d %H:%M:%S"
-                                )
-                                file.write(
-                                    f"[{timestamp} UTC] {remove_ansi_escape_codes(message)}\n"
-                                )
+                                file.write(f"{message}\n")
                         self.artifact_store._remove_previous_file_versions(
                             self.logs_uri
                         )

zenml/zen_stores/migrations/utils.py

@@ -267,7 +267,7 @@ class MigrationUtils(BaseModel):
                             # table creation statement.
                             continue
 
-                        index_create = str(CreateIndex(index)).strip()  # type: ignore[no-untyped-call]
+                        index_create = str(CreateIndex(index)).strip()
                         index_create = index_create.replace(
                             f"CREATE INDEX {index.name}",
                             f"CREATE INDEX `{index.name}`",

{zenml_nightly-0.80.1.dev20250330.dist-info → zenml_nightly-0.80.1.dev20250401.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.1.dev20250330
+Version: 0.80.1.dev20250401
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -82,7 +82,7 @@ Requires-Dist: mike (>=1.1.2,<2.0.0) ; extra == "dev"
 Requires-Dist: mkdocs (>=1.6.1,<2.0.0) ; extra == "dev"
 Requires-Dist: mkdocs-autorefs (>=1.4.0,<2.0.0) ; extra == "dev"
 Requires-Dist: mkdocs-awesome-pages-plugin (>=2.10.1,<3.0.0) ; extra == "dev"
-Requires-Dist: mkdocs-material (>=9.6.5,<10.0.0) ; extra == "dev"
+Requires-Dist: mkdocs-material (==9.6.8) ; extra == "dev"
 Requires-Dist: mkdocstrings[python] (>=0.28.1,<0.29.0) ; extra == "dev"
 Requires-Dist: mypy (==1.7.1) ; extra == "dev"
 Requires-Dist: orjson (>=3.10.0,<3.11.0) ; extra == "server"

{zenml_nightly-0.80.1.dev20250330.dist-info → zenml_nightly-0.80.1.dev20250401.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=kMQ0fv3IdpHMPnr9sR_jsNj64648-jDrzRFS1IBI5nI,19
+zenml/VERSION,sha256=cUdRWGuMppYBvD9m95WTtdZz_Cxt53WVPydhixSDg_E,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
@@ -269,7 +269,7 @@ 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=TSCl_q-T9FHCh7pPUmkb3GgsLrTjvaV5mShdVcxd0Vw,39212
+zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=vlQgHysKkg_BfIANTiZSPPlGInQYRoX4k6EYXt-ei90,39341
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
 zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=JGrFTKkQV4ZDn_yTEqX-AtBMraFasFgzLVws2mvhS64,94915
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
@@ -577,7 +577,7 @@ zenml/io/filesystem_registry.py,sha256=stujDg4a5k983WMwp3rj4Z4puiUco4REyVoIoMIpI
 zenml/io/local_filesystem.py,sha256=xQTZPT5cpooptUV8KiifxZojS6pWCv1-6UUxitUYb_E,7386
 zenml/logger.py,sha256=LMV2sMFQ-6JK9xEn6kEt1C9vAoQ0lwuHVM5XHIeKubE,6966
 zenml/logging/__init__.py,sha256=lnqbOa31wAHwPP5f8vZazOrUwnP2QviLiIVwxoAefD8,975
-zenml/logging/step_logging.py,sha256=6QDfn4oN4h7D_M3RxkVre0NBYRou8IeVtk1vyO89tqQ,18551
+zenml/logging/step_logging.py,sha256=KQN9SZO5CQSASctP8OXymBmgNyNNfDT9QkbNwGV-QKM,18272
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
 zenml/login/credentials.py,sha256=RtLmYkFQ5trbprhsO8NCRKwBA136KzGoUWY52dZP9GA,12722
 zenml/login/credentials_store.py,sha256=m5MaImmQleEi79S0ogPZIVeYvMhWcPO1UWuYns4Zr8E,23673
@@ -1072,7 +1072,7 @@ zenml/zen_stores/migrations/__init__.py,sha256=N9CHfdz0AZ6KniQ450VCIV3H0CuWtx83A
 zenml/zen_stores/migrations/alembic.py,sha256=JDqx7Md6DxnHtP3xrZG1I0cNv6NyTR0oO3tPRUPaS2I,7455
 zenml/zen_stores/migrations/env.py,sha256=hN6GqD2toKL-r9y0FFAf2seJfr79Mzaeqslh5kObcos,1730
 zenml/zen_stores/migrations/script.py.mako,sha256=wTJhgE4DA8I2iVA29sfx74WLfbi3GBnXEwGnH5nNj4s,695
-zenml/zen_stores/migrations/utils.py,sha256=QsWDm-uQNzD0RvIA54CMFnLrEUgXmXGfdQcTnwTdhYI,29560
+zenml/zen_stores/migrations/utils.py,sha256=_o6qr0_fKrcYYtrVK3kT0ZrjpyqFiMzRGsSmsqRPaZ0,29527
 zenml/zen_stores/migrations/versions/0.21.0_release.py,sha256=5FoOdsRLifIC10GQBx3C7cSthPM3wtuzO4heMXgTbcY,444
 zenml/zen_stores/migrations/versions/0.21.1_release.py,sha256=SLWTKX1s6yN2xUEDe2-9_sRMqBjJPr1Je_I7JZTf1r8,432
 zenml/zen_stores/migrations/versions/0.22.0_release.py,sha256=bxpLvwzqSin2RuT-uYK4DWl1D92ldvhJ_QVdRFub8DE,444
@@ -1310,8 +1310,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=ldyC1uhMnmX5ojnqY9d_L2S-iC-eaNUwsexTkdPtqr4,440204
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.80.1.dev20250330.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.80.1.dev20250330.dist-info/METADATA,sha256=8PjIYKQVo4eDNgQ1Xqw5QdjXzMgZhq7XpjvSlHLoyjE,24227
-zenml_nightly-0.80.1.dev20250330.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-zenml_nightly-0.80.1.dev20250330.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.80.1.dev20250330.dist-info/RECORD,,
+zenml_nightly-0.80.1.dev20250401.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.80.1.dev20250401.dist-info/METADATA,sha256=ZOZxY3_UgQAQv7q8JXVO1FHqLdWxyriYQcp5MnFeGtQ,24219
+zenml_nightly-0.80.1.dev20250401.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.80.1.dev20250401.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.80.1.dev20250401.dist-info/RECORD,,