From 28059eb5e0aad715823ee4f6509d6a9e3d6e5db0 Mon Sep 17 00:00:00 2001 From: tvayer Date: Wed, 29 May 2019 17:11:41 +0200 Subject: py2 error --- test/test_gromov.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_gromov.py b/test/test_gromov.py index ec85abf..3ca184b 100644 --- a/test/test_gromov.py +++ b/test/test_gromov.py @@ -13,7 +13,7 @@ import ot def test_gromov(): np.random.seed(42) - n_samples = 50 # nb samples + n_samples = 50.0 # nb samples mu_s = np.array([0, 0]) cov_s = np.array([[1, 0], [0, 1]]) @@ -161,7 +161,7 @@ def test_gromov_entropic_barycenter(): def test_fgw(): np.random.seed(42) - n_samples = 50 # nb samples + n_samples = 50.0 # nb samples mu_s = np.array([0, 0]) cov_s = np.array([[1, 0], [0, 1]]) -- cgit v1.2.3