summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authortvayer <titouan.vayer@gmail.com>2019-05-29 17:11:41 +0200
committertvayer <titouan.vayer@gmail.com>2019-05-29 17:11:41 +0200
commit28059eb5e0aad715823ee4f6509d6a9e3d6e5db0 (patch)
treebd07374c9b9e1aef0d7e463f75301c9c3d279d70 /test
parente1bd94bb7e85a0d2fd0fcd7642b06da12c1db6db (diff)
py2 error
Diffstat (limited to 'test')
-rw-r--r--test/test_gromov.py4
1 files 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]])