From f95d4d7489c835e9a1135c2201a78cfed7d9e7e9 Mon Sep 17 00:00:00 2001 From: eloitanguy Date: Mon, 18 Nov 2024 10:44:13 +0100 Subject: [PATCH] no longer impose ECOS solver for SSNB, use new CVXPY default (Clarabel) --- ot/mapping.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ot/mapping.py b/ot/mapping.py index dae059edb..94ac6d3fd 100644 --- a/ot/mapping.py +++ b/ot/mapping.py @@ -173,7 +173,7 @@ def nearest_brenier_potential_fit( - c3 * (G[j] - G[i]).T @ (X[j] - X[i]) ] problem = cvx.Problem(objective, constraints) - problem.solve(solver=cvx.ECOS) + problem.solve() phi_val, G_val = phi.value, G.value it_log_dict = { "solve_time": problem.solver_stats.solve_time, @@ -368,7 +368,7 @@ def nearest_brenier_potential_predict_bounds( - c3 * (G[j] - G_l_y).T @ (X[j] - Y[y_idx]) ] problem = cvx.Problem(objective, constraints) - problem.solve(solver=cvx.ECOS) + problem.solve() phi_lu[0, y_idx] = phi_l_y.value G_lu[0, y_idx] = G_l_y.value if log: @@ -395,7 +395,7 @@ def nearest_brenier_potential_predict_bounds( - c3 * (G_u_y - G[i]).T @ (Y[y_idx] - X[i]) ] problem = cvx.Problem(objective, constraints) - problem.solve(solver=cvx.ECOS) + problem.solve() phi_lu[1, y_idx] = phi_u_y.value G_lu[1, y_idx] = G_u_y.value if log: