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

zenml/VERSION

@@ -1 +1 @@
-0.74.0.dev20250211
+0.74.0.dev20250212

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_nightly-0.74.0.dev20250211.dist-info → zenml_nightly-0.74.0.dev20250212.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.74.0.dev20250211
+Version: 0.74.0.dev20250212
 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.dev20250212.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=UhG6I4o4NTBc6VFy231HJr7TSwPCzeLflYctAPsY-y8,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
@@ -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.dev20250212.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.74.0.dev20250212.dist-info/METADATA,sha256=FzxLNILSrBhOG7JLb1Gs6gaiP3LOibyLTJWhRHHKIUQ,21430
+zenml_nightly-0.74.0.dev20250212.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+zenml_nightly-0.74.0.dev20250212.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.74.0.dev20250212.dist-info/RECORD,,