zenml-nightly 0.74.0.dev20250211__py3-none-any.whl → 0.74.0.dev20250213__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.74.0.dev20250211
+0.74.0.dev20250213

zenml/models/v2/core/model_version.py

@@ -28,7 +28,7 @@ from uuid import UUID
 from pydantic import BaseModel, ConfigDict, Field, PrivateAttr, field_validator
 
 from zenml.constants import STR_FIELD_MAX_LENGTH, TEXT_FIELD_MAX_LENGTH
-from zenml.enums import ModelStages
+from zenml.enums import ArtifactType, ModelStages
 from zenml.metadata.metadata_types import MetadataType
 from zenml.models.v2.base.filter import AnyQuery
 from zenml.models.v2.base.page import Page
@@ -426,32 +426,36 @@ class ModelVersionResponse(
 
     def _get_linked_object(
         self,
-        collection: Dict[str, Dict[str, UUID]],
         name: str,
         version: Optional[str] = None,
+        type: Optional[ArtifactType] = None,
     ) -> Optional["ArtifactVersionResponse"]:
         """Get the artifact linked to this model version given type.
 
         Args:
-            collection: The collection to search in (one of
-                self.model_artifact_ids, self.data_artifact_ids,
-                self.deployment_artifact_ids)
             name: The name of the artifact to retrieve.
             version: The version of the artifact to retrieve (None for
                 latest/non-versioned)
+            type: The type of the artifact to filter by.
 
         Returns:
             Specific version of an artifact from collection or None
         """
         from zenml.client import Client
 
-        client = Client()
+        artifact_versions = Client().list_artifact_versions(
+            sort_by="desc:created",
+            size=1,
+            name=name,
+            version=version,
+            model_version_id=self.id,
+            type=type,
+            hydrate=True,
+        )
 
-        if name not in collection:
+        if not artifact_versions.items:
             return None
-        if version is None:
-            version = max(collection[name].keys())
-        return client.get_artifact_version(collection[name][version])
+        return artifact_versions.items[0]
 
     def get_artifact(
         self,
@@ -468,12 +472,7 @@ class ModelVersionResponse(
         Returns:
             Specific version of an artifact or None
         """
-        all_artifact_ids = {
-            **self.model_artifact_ids,
-            **self.data_artifact_ids,
-            **self.deployment_artifact_ids,
-        }
-        return self._get_linked_object(all_artifact_ids, name, version)
+        return self._get_linked_object(name, version)
 
     def get_model_artifact(
         self,
@@ -490,7 +489,7 @@ class ModelVersionResponse(
         Returns:
             Specific version of the model artifact or None
         """
-        return self._get_linked_object(self.model_artifact_ids, name, version)
+        return self._get_linked_object(name, version, ArtifactType.MODEL)
 
     def get_data_artifact(
         self,
@@ -507,11 +506,7 @@ class ModelVersionResponse(
         Returns:
             Specific version of the data artifact or None
         """
-        return self._get_linked_object(
-            self.data_artifact_ids,
-            name,
-            version,
-        )
+        return self._get_linked_object(name, version, ArtifactType.DATA)
 
     def get_deployment_artifact(
         self,
@@ -528,11 +523,7 @@ class ModelVersionResponse(
         Returns:
             Specific version of the deployment artifact or None
         """
-        return self._get_linked_object(
-            self.deployment_artifact_ids,
-            name,
-            version,
-        )
+        return self._get_linked_object(name, version, ArtifactType.SERVICE)
 
     def get_pipeline_run(self, name: str) -> "PipelineRunResponse":
         """Get pipeline run linked to this version.

zenml/pipelines/pipeline_definition.py

@@ -757,10 +757,16 @@ To avoid this consider setting pipeline parameters only in one place (config or
             build=build_model,
             can_download_from_code_repository=can_download_from_code_repository,
         ):
-            code_archive = code_utils.CodeArchive(
-                root=source_utils.get_source_root()
+            source_root = source_utils.get_source_root()
+            code_archive = code_utils.CodeArchive(root=source_root)
+            logger.info(
+                "Archiving pipeline code directory: `%s`. If this is taking "
+                "longer than you expected, make sure your source root "
+                "is set correctly by running `zenml init`, and that it "
+                "does not contain unnecessarily huge files.",
+                source_root,
             )
-            logger.info("Archiving pipeline code...")
+
             code_path = code_utils.upload_code_if_necessary(code_archive)
 
         request = PipelineDeploymentRequest(

zenml/utils/source_utils.py

@@ -275,6 +275,35 @@ def resolve(
     )
 
 
+def get_implicit_source_root() -> str:
+    """Get the implicit source root (the parent directory of the main module).
+
+    Raises:
+        RuntimeError: If the main module file can't be found.
+
+    Returns:
+        The implicit source root.
+    """
+    main_module = sys.modules.get("__main__")
+    if main_module is None:
+        raise RuntimeError(
+            "Unable to determine source root because the main module could not "
+            "be found."
+        )
+
+    if not hasattr(main_module, "__file__") or not main_module.__file__:
+        raise RuntimeError(
+            "Unable to determine source root because the main module does not "
+            "have an associated file. This could be because you're running in "
+            "an interactive Python environment. If you are trying to run from "
+            "within a Jupyter notebook, please run `zenml init` from the root "
+            "where your notebook is located and restart your notebook server."
+        )
+
+    path = Path(main_module.__file__).resolve().parent
+    return str(path)
+
+
 def get_source_root() -> str:
     """Get the source root.
 
@@ -286,9 +315,6 @@ def get_source_root() -> str:
 
     Returns:
         The source root.
-
-    Raises:
-        RuntimeError: If the main module file can't be found.
     """
     if _CUSTOM_SOURCE_ROOT:
         logger.debug("Using custom source root: %s", _CUSTOM_SOURCE_ROOT)
@@ -301,26 +327,12 @@ def get_source_root() -> str:
         logger.debug("Using repository root as source root: %s", repo_root)
         return str(repo_root.resolve())
 
-    main_module = sys.modules.get("__main__")
-    if main_module is None:
-        raise RuntimeError(
-            "Unable to determine source root because the main module could not "
-            "be found."
-        )
-
-    if not hasattr(main_module, "__file__") or not main_module.__file__:
-        raise RuntimeError(
-            "Unable to determine source root because the main module does not "
-            "have an associated file. This could be because you're running in "
-            "an interactive Python environment. If you are trying to run from "
-            "within a Jupyter notebook, please run `zenml init` from the root "
-            "where your notebook is located and restart your notebook server.   "
-        )
-
-    path = Path(main_module.__file__).resolve().parent
-
-    logger.debug("Using main module parent directory as source root: %s", path)
-    return str(path)
+    implicit_source_root = get_implicit_source_root()
+    logger.debug(
+        "Using main module parent directory as source root: %s",
+        implicit_source_root,
+    )
+    return implicit_source_root
 
 
 def set_custom_source_root(source_root: Optional[str]) -> None:

{zenml_nightly-0.74.0.dev20250211.dist-info → zenml_nightly-0.74.0.dev20250213.dist-info}/METADATA

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

{zenml_nightly-0.74.0.dev20250211.dist-info → zenml_nightly-0.74.0.dev20250213.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=pF1JOjzBClX_viBsCK5CGd0U5Rsb-7M7-Qbkq6qLhPc,19
+zenml/VERSION,sha256=m5UFAlfM2bHdwaLw8Ttt42SvpS5yV6jrrfa_bMCNvyA,19
 zenml/__init__.py,sha256=SkMObQA41ajqdZqGErN00S1Vf3KAxpLvbZ-OBy5uYoo,2130
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -639,7 +639,7 @@ zenml/models/v2/core/event_source_flavor.py,sha256=TgrGkypnJUutMvvHE7Y9MbUiEeNXZ
 zenml/models/v2/core/flavor.py,sha256=tcn2GDbkOyYjXTor_7_kR7FuLwcXjlfzvAV2rU1NVM4,12806
 zenml/models/v2/core/logs.py,sha256=UhaPOTJD7IFKH8YozsaVgXYNNDHre8amheIm5JMZYQU,5824
 zenml/models/v2/core/model.py,sha256=yok73T2D4hzYGP0nhDs5mq-1LKA9iol_k5DQyCVwf1o,11918
-zenml/models/v2/core/model_version.py,sha256=q3fTYRSLR7i_84VFWqCmcm4FqGhwtlq2Wfhtm6G0u8M,21062
+zenml/models/v2/core/model_version.py,sha256=fkERxxFR_08LosUdbPF5hIeoz0n4C0703f5k_KE9mxQ,20807
 zenml/models/v2/core/model_version_artifact.py,sha256=ZyupuEsfPSjgaHDIUOojwJepJt1LTXDoUyE2h9htB_s,9086
 zenml/models/v2/core/model_version_pipeline_run.py,sha256=JbPZZEQvOK9I32htkWdAONy7gvtoz_Jo45B5gYrh2vs,7036
 zenml/models/v2/core/pipeline.py,sha256=cql05iz2jqrlyRfn3jB4iQ_AgpHapD6Sr5Kd_6Q1W30,12044
@@ -695,7 +695,7 @@ zenml/pipelines/__init__.py,sha256=hpIX7hN8jsQRHT5R-xSXZL88qrHwkmrvGLQeu1rWt4o,8
 zenml/pipelines/build_utils.py,sha256=DkID1YnRYkw569uU-gidpF8WR8E7K55_wn3CzRPQ3Cs,27562
 zenml/pipelines/pipeline_context.py,sha256=V_p-F9W7cBIlTjS0iv5-uJYMzaOj8bAUkc_uNhQgBms,3579
 zenml/pipelines/pipeline_decorator.py,sha256=FIbflYOMavbuyGmqsx3F5zZgg0oXMTi1eAcGXciljOs,4293
-zenml/pipelines/pipeline_definition.py,sha256=Pi4OGs7-WXbsm8_zczY5lpQEyLJPqkKZHjXAKfnk3HY,56936
+zenml/pipelines/pipeline_definition.py,sha256=pl5tV4_h03MQDaIJZQCJJ8mY5S3Js-89r5NFIBpxB6A,57244
 zenml/pipelines/run_utils.py,sha256=4KuHIQFtLXTZNQBScTEkIG5pqNtu6xGm6UZT7ptyyKs,11623
 zenml/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zenml/plugins/base_plugin_flavor.py,sha256=88IxFW91UB_rQ8xPlfRnIhIJh7A308NEq2epMMdlOng,2530
@@ -787,7 +787,7 @@ zenml/utils/secret_utils.py,sha256=gEvqnhzAZwPO6mdOQWvioeH-xLoSObfaNRzt17N8zyU,5
 zenml/utils/settings_utils.py,sha256=lAK13CiDCDkcLygizDbWB9q-9ukteVBJPypzFCrne9k,4631
 zenml/utils/singleton.py,sha256=uFRrUlUdS5VyY9lLJyl_n5kqppsqJLKkBhSj4g5VPkY,2757
 zenml/utils/source_code_utils.py,sha256=8iyNA2MGIORYVEkSdxNTXfS1ZdFKXTAG1dZRkeQtPL0,3751
-zenml/utils/source_utils.py,sha256=ow3fbodjmAfpwmUXBSg3oyibHk065aSvi4UV9uJD6Yk,26736
+zenml/utils/source_utils.py,sha256=joKLghhDq9dh0fd8B0WRGX-nN-uwnGQdgmsyY_n-8gY,27033
 zenml/utils/string_utils.py,sha256=VPuAwFzxwXXo1cKmBWxHnoeTyDX35XPf_lXGiDMXPEw,7254
 zenml/utils/time_utils.py,sha256=-9Y9zwJ-6Gv7hoZQCoftPyC2LCLo2bYj6OgdyBaE44o,4076
 zenml/utils/typed_model.py,sha256=00EAo1I1VnOBHG4-ce8dPkyHRPpgi67SRIU-KdewRWs,4757
@@ -1280,8 +1280,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=tKOghX2Wa0f0xSYu_ReqHgCwslvHrBhE3RePJG2tcA0,415921
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.74.0.dev20250211.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.74.0.dev20250211.dist-info/METADATA,sha256=EXuTz3dmmjR05PaCaBnMmGmMzhE7XphPSKhA1JIrnbI,21430
-zenml_nightly-0.74.0.dev20250211.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-zenml_nightly-0.74.0.dev20250211.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.74.0.dev20250211.dist-info/RECORD,,
+zenml_nightly-0.74.0.dev20250213.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.74.0.dev20250213.dist-info/METADATA,sha256=FfHRQAkvQcU-SpjOdUwnsKp1KURqE1eBLVm5L6Pd67M,21430
+zenml_nightly-0.74.0.dev20250213.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+zenml_nightly-0.74.0.dev20250213.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.74.0.dev20250213.dist-info/RECORD,,