Fix two more uniqueness constraints

- SqlExperimentPermission: "experiment_id", "user_id"
- SqlRegisteredModelPermission: "name", "user_id"

CHANGES.md

@@ -3,7 +3,10 @@
 
 ## in progress
 - Update to MLflow 2.8.0
-- Fix uniqueness constraint with `mlflow.server.auth.db.models.SqlUser.username`.
+- Fix uniqueness constraints
+ - `mlflow.server.auth.db.models.SqlUser.username`.
+ - `m.s.a.d.m.SqlExperimentPermission`: "experiment_id", "user_id"
+ - `m.s.a.d.m.SqlRegisteredModelPermission`: "name", "user_id"
 
 ## 2023-11-01 v2.7.1
 - Fix uniqueness constraint with `SqlRegisteredModel.name`. Thanks, @andnig.

mlflow_cratedb/patch/mlflow/model.py

@@ -14,12 +14,24 @@ def polyfill_uniqueness_constraints():
  - SqlExperimentPermission: "experiment_id", "user_id"
  - SqlRegisteredModelPermission: "name", "user_id"
  """
- from mlflow.server.auth.db.models import SqlUser
+ from mlflow.server.auth.db.models import SqlExperimentPermission, SqlRegisteredModelPermission, SqlUser
  from mlflow.store.model_registry.dbmodels.models import SqlRegisteredModel
  from mlflow.store.tracking.dbmodels.models import SqlExperiment
 
  listen(SqlExperiment, "before_insert", check_uniqueness_factory(SqlExperiment, "name"))
+ listen(
+ SqlExperimentPermission,
+ "before_insert",
+ check_uniqueness_factory(SqlExperimentPermission, "experiment_id", "user_id"),
+ )
+
  listen(SqlRegisteredModel, "before_insert", check_uniqueness_factory(SqlRegisteredModel, "name"))
+ listen(
+ SqlRegisteredModelPermission,
+ "before_insert",
+ check_uniqueness_factory(SqlRegisteredModelPermission, "name", "user_id"),
+ )
+
  listen(SqlUser, "before_insert", check_uniqueness_factory(SqlUser, "username"))
 
 

pyproject.toml

@@ -78,7 +78,7 @@ dynamic = [
 dependencies = [
  "crash",
  "crate[sqlalchemy]>=0.34",
- "cratedb-toolkit==0.0.1",
+ "cratedb-toolkit==0.0.2",
  "mlflow==2.8",
  "sqlparse<0.5",
 ]
@@ -93,8 +93,8 @@ develop = [
  "validate-pyproject<0.16",
 ]
 examples = [
+ "pycaret[analysis,models,parallel,test,tuner]==3.1",
  "salesforce-merlion<2.1",
- "pycaret[analysis,models,tuner,parallel,test]==3.1.0",
  "werkzeug==2.2.3",
 ]
 release = [