zenml-nightly 0.80.1.dev20250331__py3-none-any.whl → 0.80.1.dev20250402__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.80.1.dev20250331
+0.80.1.dev20250402

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/integrations/mlflow/__init__.py

@@ -63,7 +63,7 @@ class MlflowIntegration(Integration):
         
 
         reqs = [
-            "mlflow>=2.1.1,<2.21.0",
+            "mlflow>=2.1.1,<3",
             # TODO: remove this requirement once rapidjson is fixed
             "python-rapidjson<1.15",
             # When you do:

zenml/integrations/mlflow/model_registries/mlflow_model_registry.py

@@ -13,8 +13,10 @@
 #  permissions and limitations under the License.
 """Implementation of the MLflow model registry for ZenML."""
 
+import os
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Tuple, cast
+from urllib.parse import unquote, urlparse
 
 import mlflow
 from mlflow import MlflowClient
@@ -47,6 +49,51 @@ from zenml.stack.stack_validator import StackValidator
 logger = get_logger(__name__)
 
 
+def _remove_file_scheme(uri: str) -> str:
+    """Parse a URI and remove the file:// scheme if present.
+
+    Args:
+        uri: The URI to remove the file scheme from.
+
+    Returns:
+        The decoded file path, preserving network hostnames.
+    """
+    # Parse the URI
+    parsed = urlparse(uri)
+
+    # Check if it uses the file scheme
+    if parsed.scheme == "file":
+        # Get the path and handle any URL encoding
+        path = unquote(parsed.path)
+
+        # Handle network paths with hostname
+        if parsed.netloc:
+            # Create a proper network path with the hostname
+            network_path = f"//{parsed.netloc}{path}"
+
+            # On Windows, you might want to convert to backslashes
+            if os.name == "nt":
+                network_path = network_path.replace("/", "\\")
+
+            return network_path
+
+        # Handle local paths (no hostname)
+        else:
+            # On Windows, file:///C:/path becomes /C:/path after parsing
+            if (
+                path.startswith("/")
+                and len(path) > 1
+                and path[2] == ":"
+                and path[1].isalpha()
+            ):
+                # This is a Windows path with a drive letter
+                return path[1:]  # Remove the leading slash
+            return path
+    else:
+        # Not a file URI, return the original
+        return uri
+
+
 class MLFlowModelRegistry(BaseModelRegistry):
     """Register models using MLflow."""
 
@@ -737,7 +784,9 @@ class MLFlowModelRegistry(BaseModelRegistry):
             from mlflow.models import get_model_info
 
             model_library = (
-                get_model_info(model_uri=mlflow_model_version.source)
+                get_model_info(
+                    model_uri=_remove_file_scheme(mlflow_model_version.source)
+                )
                 .flavors.get("python_function", {})
                 .get("loader_module")
             )

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.dev20250331.dist-info → zenml_nightly-0.80.1.dev20250402.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.1.dev20250331
+Version: 0.80.1.dev20250402
 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.dev20250331.dist-info → zenml_nightly-0.80.1.dev20250402.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=SFVku8eYE7b6JtekWw-CrBzayMB2rxLl0XADXARs-OM,19
+zenml/VERSION,sha256=30n7-vAhVZVjGkiw9YleCH5MTMe4aMeJsNA_LTaoeMc,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
@@ -379,7 +379,7 @@ zenml/integrations/llama_index/__init__.py,sha256=go2bEZ1dzjUtgFWb22op3MABRBLezy
 zenml/integrations/llama_index/materializers/__init__.py,sha256=OEtWarp07nDpbSnV5Y9f8Gk1-Ufa7AINiz4e7H22rDQ,963
 zenml/integrations/llama_index/materializers/document_materializer.py,sha256=NqSEP4YbaAr8har4dGFARG7EX2Tr_Gky4-sEsKGwFAE,2355
 zenml/integrations/llama_index/materializers/gpt_index_materializer.py,sha256=4CL9f_kGrK8zAlo4K03xKOomZRNd3SQeISwLOG7J1G4,4774
-zenml/integrations/mlflow/__init__.py,sha256=TgCTmNZBbxGB2nIKmuuB676FN6m4X6TChMfkPLg3J-A,4207
+zenml/integrations/mlflow/__init__.py,sha256=eQ7mapet2-CDBlVnmSgqUHM95Li10o8f6PI8rKkU7Rk,4202
 zenml/integrations/mlflow/experiment_trackers/__init__.py,sha256=foDnjpi4vkH9adjaA01c-utb0mRYybQfdR75PDK9CAQ,775
 zenml/integrations/mlflow/experiment_trackers/mlflow_experiment_tracker.py,sha256=OhvcZgGfawv_sN4zs1I68jFgvWmrPTfC7gQ-Ldj8PHY,14315
 zenml/integrations/mlflow/flavors/__init__.py,sha256=hMKgndBdmMghG9d3o4sJNVXG4mrZiTcA6hBeL0BirOY,1305
@@ -390,7 +390,7 @@ zenml/integrations/mlflow/mlflow_utils.py,sha256=TQYeR9THJQR7tsoB17QuP2aTvxdZKUq
 zenml/integrations/mlflow/model_deployers/__init__.py,sha256=QqeQKA2KHOEFKkn_ucVprSACsNuBc-UoB51kmEnOKNs,813
 zenml/integrations/mlflow/model_deployers/mlflow_model_deployer.py,sha256=s2R5yTszazfzbI8GlOPZrKH-FyN1MHT0XtxUj_nOw9Q,10191
 zenml/integrations/mlflow/model_registries/__init__.py,sha256=vxR8IsJjB2u7z-_1MP0csZarp5J0OwGMvYngxN99_cs,813
-zenml/integrations/mlflow/model_registries/mlflow_model_registry.py,sha256=dQKeTqATyC_lsJxAWiSTsJaMBzc9chjeG5zfv3RyFMQ,26867
+zenml/integrations/mlflow/model_registries/mlflow_model_registry.py,sha256=rttQL7RNMdEzU5GaElE_Y8Oq9cv2O96OHO-h8IiKr3k,28347
 zenml/integrations/mlflow/services/__init__.py,sha256=GnXApvQbsBU0M303i58-G3JDGnuDhJ0EuHrmPdvkzJo,791
 zenml/integrations/mlflow/services/mlflow_deployment.py,sha256=kFouyBcK8wYfzm_QW6si9add3WyftqFKLvth3aGsXJI,11141
 zenml/integrations/mlflow/steps/__init__.py,sha256=5IXeipGRfBjtqr0ZdbQLliuQNr5GXsm7xkhpqfOg6qI,770
@@ -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.dev20250331.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.80.1.dev20250331.dist-info/METADATA,sha256=DTVmnbjnu43gscMeYIasIsZ_ytEs5zHpFE8pDFfIBf4,24227
-zenml_nightly-0.80.1.dev20250331.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-zenml_nightly-0.80.1.dev20250331.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.80.1.dev20250331.dist-info/RECORD,,
+zenml_nightly-0.80.1.dev20250402.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.80.1.dev20250402.dist-info/METADATA,sha256=zl1mde_15G5DhcxZNKCvzi_K-4sdukCQARSLvrol_c4,24219
+zenml_nightly-0.80.1.dev20250402.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.80.1.dev20250402.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.80.1.dev20250402.dist-info/RECORD,,