complete solve_gromov

Author: cedricvincentcuaz

ot/solvers.py

@@ -863,7 +863,8 @@ def solve_gromov(
 
     if reg is None or reg == 0:  # exact OT
         if unbalanced is None and unbalanced_type.lower() not in [
-            "semirelaxed"
+            "semirelaxed",
+            "partial",
         ]:  # Exact balanced OT
             if M is None or alpha == 1:  # Gromov-Wasserstein problem
                 # default values for solver

test/test_solvers.py

@@ -518,6 +518,10 @@ def test_solve_gromov_not_implemented(nx):
         ot.solve_gromov(Ca, Cb, unbalanced_type="partial", unbalanced=1.5)
     with pytest.raises(ValueError):
         ot.solve_gromov(Ca, Cb, reg=1, unbalanced_type="partial", unbalanced=1.5)
+    with pytest.raises(ValueError):
+        ot.solve_gromov(Ca, Cb, M, unbalanced_type="partial", unbalanced=1.5)
+    with pytest.raises(ValueError):
+        ot.solve_gromov(Ca, Cb, M, reg=1, unbalanced_type="partial", unbalanced=1.5)
 
 
 def test_solve_sample(nx):