summaryrefslogtreecommitdiff
path: root/test/test_bregman.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_bregman.py')
-rw-r--r--test/test_bregman.py1
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