diff options
Diffstat (limited to 'test/test_bregman.py')
-rw-r--r-- | test/test_bregman.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index 331acd3..1ebd21f 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -321,8 +321,9 @@ def test_implemented_methods(): # make dists unbalanced b = ot.utils.unif(n) A = rng.rand(n, 2) + A /= A.sum(0, keepdims=True) M = ot.dist(x, x) - epsilon = 1. + epsilon = 1.0 for method in IMPLEMENTED_METHODS: ot.bregman.sinkhorn(a, b, M, epsilon, method=method) |