mlflow-tclake-plugin 2.1.7.dev2__tar.gz → 2.1.7.dev5__tar.gz

{mlflow_tclake_plugin-2.1.7.dev2 → mlflow_tclake_plugin-2.1.7.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlflow-tclake-plugin
-Version: 2.1.7.dev2
+Version: 2.1.7.dev5
 Summary: Tclake plugin for MLflow
 License-File: LICENSE.txt
 Requires-Dist: mlflow==2.17.2

{mlflow_tclake_plugin-2.1.7.dev2 → mlflow_tclake_plugin-2.1.7.dev5}/mlflow_tclake_plugin/tclake_store.py

@@ -71,6 +71,9 @@ TCLAKE_MLFLOW_MODEL_SIGNATURE_KEY = "tclake.mlflow.model_signature"
 TCLAKE_WEDATA_PROJECT_ID_KEY = "wedata.project"
 TCLAKE_MLFLOW_MODEL_ARTIFACT_PATH_KEY = "tclake.mlflow.artifact_path"
 WEDATA_DATASCIENCE_TYPE = "wedata.datascience.type"
+# 这些内置 key 在写入侧被存成无前缀属性（见 _add_tags_to_properties /
+# _add_project_id_to_properties），读回时需要一并还原成 tag，否则调用方拿不到。
+UNPREFIXED_BUILTIN_TAG_KEYS = (WEDATA_DATASCIENCE_TYPE, TCLAKE_WEDATA_PROJECT_ID_KEY)
 
 
 def _set_kv_to_properties(key, value, properties=None):
@@ -216,9 +219,19 @@ def _get_tags_from_properties(properties):
     if properties is None:
         return None
     tags = []
+    seen_keys = set()
+    # 优先取带前缀的属性：调用方通过 tags 参数显式写入的值
     for p in properties:
         if p["Key"].startswith(TCLAKE_MLFLOW_TAG_PREFIX):
-            tags.append(ModelVersionTag(p["Key"][len(TCLAKE_MLFLOW_TAG_PREFIX):], p["Value"]))
+            tag_key = p["Key"][len(TCLAKE_MLFLOW_TAG_PREFIX):]
+            if tag_key not in seen_keys:
+                seen_keys.add(tag_key)
+                tags.append(ModelVersionTag(tag_key, p["Value"]))
+    # 再补无前缀存储的内置 key，已由带前缀属性覆盖的不重复添加
+    for p in properties:
+        if p["Key"] in UNPREFIXED_BUILTIN_TAG_KEYS and p["Key"] not in seen_keys:
+            seen_keys.add(p["Key"])
+            tags.append(ModelVersionTag(p["Key"], p["Value"]))
     return tags
 
 
@@ -253,6 +266,8 @@ def _get_model_version_name(entity):
 def _make_model_version(entity, name):
     properties = entity["Properties"]
     audit = entity["Audit"]
+    # 部分接口响应可能不含 Aliases 字段，缺失时按空列表处理
+    aliases = entity.get("Aliases") or []
     return ModelVersion(
         name=name,
         version=_get_model_version(entity["Version"]),
@@ -263,6 +278,7 @@ def _make_model_version(entity, name):
         run_id=_get_run_id_from_properties(properties),
         tags=_get_tags_from_properties(properties),
         run_link=_get_run_link_from_properties(properties),
+        aliases=aliases,
         status="READY",
     )
 

{mlflow_tclake_plugin-2.1.7.dev2 → mlflow_tclake_plugin-2.1.7.dev5}/mlflow_tclake_plugin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mlflow-tclake-plugin
-Version: 2.1.7.dev2
+Version: 2.1.7.dev5
 Summary: Tclake plugin for MLflow
 License-File: LICENSE.txt
 Requires-Dist: mlflow==2.17.2

{mlflow_tclake_plugin-2.1.7.dev2 → mlflow_tclake_plugin-2.1.7.dev5}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="mlflow-tclake-plugin",
-    version="2.1.7.dev2",
+    version="2.1.7.dev5",
     description="Tclake plugin for MLflow",
     packages=find_packages(),
     # Require MLflow as a dependency of the plugin, so that plugin users can simply install