diff options
author | Laetitia Chapel <laetitia.chapel@univ-ubs.fr> | 2020-10-22 10:58:31 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-10-22 10:58:31 +0200 |
commit | 93785eba11b59d544f1edde6661e93ee587148ee (patch) | |
tree | d3177072f3a27c194136eda86bf5bd63e4297e82 /test | |
parent | 78b44af2434f494c8f9e4c8c91003fbc0e1d4415 (diff) |
[MRG] Fix bugs for partial OT (#215)
* bugfix
* update refs partial OT
* fixes small typos in plot_partial_wass_and_gromov
* fix small bugs in partial.py
* update README
* pep8 bugfix
* modif doctest
* fix bugtests
* update on test_partial and test on the numerical precision on ot/partial
* resolve merge pb
Co-authored-by: RĂ©mi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'test')
-rwxr-xr-x | test/test_partial.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/test_partial.py b/test/test_partial.py index 510e081..121f345 100755 --- a/test/test_partial.py +++ b/test/test_partial.py @@ -51,10 +51,12 @@ def test_raise_errors(): ot.partial.partial_gromov_wasserstein(M, M, p, q, m=-1, log=True) with pytest.raises(ValueError): - ot.partial.entropic_partial_gromov_wasserstein(M, M, p, q, reg=1, m=2, log=True) + ot.partial.entropic_partial_gromov_wasserstein(M, M, p, q, reg=1, m=2, + log=True) with pytest.raises(ValueError): - ot.partial.entropic_partial_gromov_wasserstein(M, M, p, q, reg=1, m=-1, log=True) + ot.partial.entropic_partial_gromov_wasserstein(M, M, p, q, reg=1, m=-1, + log=True) def test_partial_wasserstein_lagrange(): |