From ba493aa5488507937b7f9707faa17128c9aa1872 Mon Sep 17 00:00:00 2001 From: ievred Date: Tue, 31 Mar 2020 17:36:00 +0200 Subject: readme move to bregman --- 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 a8c258a..958df7b 100644 --- a/test/test_da.py +++ b/test/test_da.py @@ -8,6 +8,7 @@ import numpy as np from numpy.testing import assert_allclose, assert_equal import ot +from ot.bregman import jcpot_barycenter from ot.datasets import make_data_classif from ot.utils import unif @@ -603,7 +604,6 @@ def test_jcpot_transport_class(): # test transform transp_Xs = otda.transform(Xs=Xs) [assert_equal(x.shape, y.shape) for x, y in zip(transp_Xs, Xs)] - #assert_equal(transp_Xs.shape, Xs.shape) Xs_new, _ = make_data_classif('3gauss', ns1 + 1) transp_Xs_new = otda.transform(Xs_new) -- cgit v1.2.3