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

zenml/VERSION

@@ -1 +1 @@
-0.80.1.dev20250401
+0.80.1.dev20250403

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_nightly-0.80.1.dev20250401.dist-info → zenml_nightly-0.80.1.dev20250403.dist-info}/METADATA

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

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

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=cUdRWGuMppYBvD9m95WTtdZz_Cxt53WVPydhixSDg_E,19
+zenml/VERSION,sha256=aWwg-fBqgZ9oDR6dhHJc5o2EySW_O7NhHPKAZD1QVv4,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
@@ -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
@@ -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.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,,
+zenml_nightly-0.80.1.dev20250403.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.80.1.dev20250403.dist-info/METADATA,sha256=p2L6CBN_1i1UQ4BoPZS7fAbYx6DR-8cXP4TQcsZ3TQQ,24219
+zenml_nightly-0.80.1.dev20250403.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+zenml_nightly-0.80.1.dev20250403.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.80.1.dev20250403.dist-info/RECORD,,