mlflow-tclake-plugin 2.1.6__tar.gz → 2.1.7.dev2__tar.gz

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

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

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

@@ -365,7 +365,7 @@ class TCLakeStore(AbstractStore):
         resp = self._call("DescribeModelVersions", req_body)
         model_version = None
         for mv in resp["ModelVersions"]:
-            if alias in mv["Aliases"]:
+            if alias in (mv.get("Aliases") or []):
                 model_version = mv
                 break
         if model_version is None:
@@ -713,10 +713,42 @@ class TCLakeStore(AbstractStore):
         self._call("ModifyModelVersionProperties", req)
 
     def set_registered_model_alias(self, name, alias, version):
-        raise NotImplementedError("Method not implemented")
+        log_msg("set_registered_model_alias {} {} {}".format(name, alias, version))
+        # 同一模型下别名唯一，若已绑定到某个版本需先删除再重设
+        exist_model_version = self._get_model_version_by_alias(name, alias)
+        if exist_model_version is not None:
+            raise MlflowException(
+                "Alias {} already exists for model {}:{}, please delete it first".format(
+                    alias, name, exist_model_version.version),
+                RESOURCE_ALREADY_EXISTS,
+            )
+        [catalog_name, schema_name, model_name] = self._split_model_name(name)
+        req = {
+            "CatalogName": catalog_name,
+            "SchemaName": schema_name,
+            "ModelName": model_name,
+            "ModelVersion": _set_model_version(version),
+            "AddedAliases": [alias],
+        }
+        self._call("ModifyModelVersionAliases", req)
 
     def delete_registered_model_alias(self, name, alias):
-        raise NotImplementedError("Method not implemented")
+        log_msg("delete_registered_model_alias {} {}".format(name, alias))
+        model_version = self._get_model_version_by_alias(name, alias)
+        if model_version is None:
+            raise MlflowException(
+                "Alias {} does not exist for model {}".format(alias, name),
+                INVALID_PARAMETER_VALUE,
+            )
+        [catalog_name, schema_name, model_name] = self._split_model_name(name)
+        req = {
+            "CatalogName": catalog_name,
+            "SchemaName": schema_name,
+            "ModelName": model_name,
+            "ModelVersion": _set_model_version(model_version.version),
+            "RemovedAliases": [alias],
+        }
+        self._call("ModifyModelVersionAliases", req)
 
     def get_model_version_by_alias(self, name, alias):
         log_msg("get_model_version_by_alias {} {}".format(name, alias))

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

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

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

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