From b028f0ce532923a2c31d40dfb1e8ba5b27422140 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Flamary?= Date: Mon, 22 Jun 2026 09:39:21 +0200 Subject: [PATCH 1/2] remove tf generig reg --- ot/solvers.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/ot/solvers.py b/ot/solvers.py index 88cf5c7ab..b84d4b1d5 100644 --- a/ot/solvers.py +++ b/ot/solvers.py @@ -557,6 +557,11 @@ def solve( isinstance(reg_type, tuple) or reg_type.lower() in ["kl", "l2", "entropy"] ) and unbalanced_type.lower() in ["kl", "l2", "tv"]: + if isinstance(reg_type, tuple) and nx.__name__ == "tf": + raise NotImplementedError( + "Custom regularization is not implemented for TensorFlow backend." + ) + if max_iter is None: max_iter = 1000 if tol is None: From 191d5f92e2ee0b3f73e12b689ce0b056c330c9d1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Flamary?= Date: Mon, 22 Jun 2026 09:42:23 +0200 Subject: [PATCH 2/2] add info to reelase file --- RELEASES.md | 1 + 1 file changed, 1 insertion(+) diff --git a/RELEASES.md b/RELEASES.md index c7024ad04..47c1f19a0 100644 --- a/RELEASES.md +++ b/RELEASES.md @@ -50,6 +50,7 @@ This new release adds support for sparse cost matrices and a new lazy EMD solver - Debug Debug linux test core dump (PR #815) - Fix entropic regularization in `gcg`(PR #817, Issue #758) - Fix documentation build on master with submodules (PR #818) +- Fix failing test for unbalanced solver with generic regularization (PR #824) ## 0.9.6.post1