fix(artschema.spec): fix entity api property in LoadArtifactSpec

lariel-fernandes · Oct 23, 2024 · 75fa4e5 · 75fa4e5
1 parent e5012fe
commit 75fa4e5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/mlopus/artschema/specs.py b/src/mlopus/artschema/specs.py
@@ -199,6 +199,11 @@ class LoadArtifactSpec(MlflowApiMixin, Generic[T, LA]):
 
     _parse_subject = pydantic.validator("subject", pre=True, allow_reuse=True)(_parse_subject)
 
+    @property
+    def entity_api(self) -> T:
+        """Entity metadata with MLFlow API handle."""
+        return self.subject.using(self.mlflow_api).entity_api
+
     def download(self) -> Path:
         """Cache subject metadata and artifact.
 
@@ -314,11 +319,6 @@ class LogArtifactSpec(MlflowApiMixin, Generic[T, LA]):
 
     _parse_subject = pydantic.validator("subject", pre=True, allow_reuse=True)(_parse_subject)
 
-    @property
-    def entity_api(self) -> T:
-        """Entity metadata with MLFlow API handle."""
-        return self.subject.using(self.mlflow_api).entity_api
-
     def log(self, artifact: A | dict | Path, schema: Schema[A, D, L] | Type[Schema[A, D, L]] | str | None = None) -> T:
         """Log artifact.
 

diff --git a/uv.lock b/uv.lock