diff options
Diffstat (limited to 'test/test_bregman.py')
-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 |