diff options
author | Mokhtar Z. Alaya <mzalaya@sct-18-1bee-1581.univ-rouen.fr> | 2020-01-16 16:44:40 +0100 |
---|---|---|
committer | Mokhtar Z. Alaya <mzalaya@sct-18-1bee-1581.univ-rouen.fr> | 2020-01-16 16:44:40 +0100 |
commit | 4918d2c619aaa654c524c9c5dc7f4dc82b838f82 (patch) | |
tree | eb452a8982763e5c7a1b84f8d45dd18c2b5fcc1d /test/test_bregman.py | |
parent | 18242437e73aba9cf131fafc1571e376b57f25f6 (diff) |
update readme
Diffstat (limited to 'test/test_bregman.py')
-rw-r--r-- | test/test_bregman.py | 3 |
1 files changed, 2 insertions, 1 deletions
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) |