diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2022-08-17 17:36:20 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-08-17 17:36:20 +0200 |
| commit | a24ec2554d003778de2c3e0550a214f83b984fa9 (patch) | |
| tree | 22fb2ead9d78719274ac4539f69a87b7f1f9c300 /test | |
| parent | f2aaf401192dd1a9a14ee273d58466b5468f30a8 (diff) | |
[MRG] Debug sinkhorn divergence bug and add proper test (#394)
* skip tets if not torch installed
* update release.md
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_bregman.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index c674da6..0f47c3f 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -879,6 +879,7 @@ def test_empirical_sinkhorn_divergence(nx): ot.bregman.empirical_sinkhorn_divergence(X_sb, X_tb, 1, a=ab, b=bb, log=True) +@pytest.mark.skipif(not torch, reason="No torch available") def test_empirical_sinkhorn_divergence_gradient(): # Test sinkhorn divergence n = 10 |