From a24ec2554d003778de2c3e0550a214f83b984fa9 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Wed, 17 Aug 2022 17:36:20 +0200
Subject: [MRG] Debug sinkhorn divergence bug and add proper test (#394)

* skip tets if not torch installed

* update release.md
---
 test/test_bregman.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'test')

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
-- 
cgit v1.2.3