From 54a129f8f17cbdbfa03c3caa296f99423536cc32 Mon Sep 17 00:00:00 2001 From: ievred Date: Wed, 15 Apr 2020 11:20:14 +0200 Subject: fix jcpot_bary test --- test/test_da.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'test') diff --git a/test/test_da.py b/test/test_da.py index 70296bf..472dc19 100644 --- a/test/test_da.py +++ b/test/test_da.py @@ -685,7 +685,7 @@ def test_jcpot_barycenter(): Xs = [Xs1, Xs2] ys = [ys1, ys2] - _, prop, = ot.bregman.jcpot_barycenter(Xs, ys, Xt, reg=.5, metric='sqeuclidean', + prop = ot.bregman.jcpot_barycenter(Xs, ys, Xt, reg=.5, metric='sqeuclidean', numItermax=10000, stopThr=1e-9, verbose=False, log=False) np.testing.assert_allclose(prop, [1 - pt, pt], rtol=1e-3, atol=1e-3) -- cgit v1.2.3