diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2021-04-19 15:03:57 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-04-19 15:03:57 +0200 |
| commit | cd3ce6140d7a2dbe2bcf05927a8dd8289f4ce9e2 (patch) | |
| tree | e39a2b00709c46c8b00772d1218f53fe33e59e11 /test/test_bregman.py | |
| parent | 2a3f2241951ea9cc044b4fba8a382b6ae9630513 (diff) | |
[MRG] Cleanup test warnings (#242)
* remove warnings in tests from docstrings
* working tets for bregman implemneted methods
* pep8
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) |