summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMokhtar Z. Alaya <mzalaya@sct-18-1bee-1581.univ-rouen.fr>2020-01-16 16:44:40 +0100
committerMokhtar Z. Alaya <mzalaya@sct-18-1bee-1581.univ-rouen.fr>2020-01-16 16:44:40 +0100
commit4918d2c619aaa654c524c9c5dc7f4dc82b838f82 (patch)
treeeb452a8982763e5c7a1b84f8d45dd18c2b5fcc1d
parent18242437e73aba9cf131fafc1571e376b57f25f6 (diff)
update readme
-rw-r--r--README.md2
-rw-r--r--test/test_bregman.py3
2 files changed, 3 insertions, 2 deletions
diff --git a/README.md b/README.md
index 987adf1..c115776 100644
--- a/README.md
+++ b/README.md
@@ -256,4 +256,4 @@ You can also post bug reports and feature requests in Github issues. Make sure t
[25] Frogner C., Zhang C., Mobahi H., Araya-Polo M., Poggio T. (2015). [Learning with a Wasserstein Loss](http://cbcl.mit.edu/wasserstein/) Advances in Neural Information Processing Systems (NIPS).
-[26] Alaya M. Z., Bérar M., Gasso G., Rakotomamonjy A. (2019). [Screening Sinkhorn Algorithm for Regularized Optimal Transport](https://papers.nips.cc/paper/9386-screening-sinkhorn-algorithm-for-regularized-optimal-transport), Advances in Neural Information Processing Systems 33 (NIPS).
+[26] Alaya M. Z., Bérar M., Gasso G., Rakotomamonjy A. (2019). [Screening Sinkhorn Algorithm for Regularized Optimal Transport](https://papers.nips.cc/paper/9386-screening-sinkhorn-algorithm-for-regularized-optimal-transport), Advances in Neural Information Processing Systems 33 (NeurIPS).
diff --git a/test/test_bregman.py b/test/test_bregman.py
index bc8f6ae..52e9fb2 100644
--- a/test/test_bregman.py
+++ b/test/test_bregman.py
@@ -338,6 +338,7 @@ def test_implemented_methods():
with pytest.raises(ValueError):
ot.bregman.sinkhorn2(a, b, M, epsilon, method=method)
+
def test_screenkhorn():
# test screenkhorn
rng = np.random.RandomState(0)
@@ -347,4 +348,4 @@ def test_screenkhorn():
x = rng.randn(n, 2)
M = ot.dist(x, x)
- G_screen = ot.bregman.screenkhorn(a, b, M, 1e-2, uniform=True, verbose=True) \ No newline at end of file
+ G_screen = ot.bregman.screenkhorn(a, b, M, 1e-2, uniform=True, verbose=True)