diff options
author | Gard Spreemann <gspr@nonempty.org> | 2020-07-09 08:50:11 +0200 |
---|---|---|
committer | Gard Spreemann <gspr@nonempty.org> | 2020-07-09 08:50:11 +0200 |
commit | 616c6039614e96b6cfd88eb7db25ce11c7302c30 (patch) | |
tree | a745ec24604f660b256dbaee214affa497c9c21e /test/test_stochastic.py | |
parent | d62f05daf348b4e554056f298c66cbd64f5e3c6e (diff) | |
parent | a16b9471d7114ec08977479b7249efe747702b97 (diff) |
Merge branch 'dfsg/latest' into debian/sid
Diffstat (limited to 'test/test_stochastic.py')
-rw-r--r-- | test/test_stochastic.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py index f0f3fc8..155622c 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -70,8 +70,8 @@ def test_stochastic_asgd(): M = ot.dist(x, x) - G = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "asgd", - numItermax=numItermax) + G, log = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "asgd", + numItermax=numItermax, log=True) # check constratints np.testing.assert_allclose( @@ -145,8 +145,8 @@ def test_stochastic_dual_sgd(): M = ot.dist(x, x) - G = ot.stochastic.solve_dual_entropic(u, u, M, reg, batch_size, - numItermax=numItermax) + G, log = ot.stochastic.solve_dual_entropic(u, u, M, reg, batch_size, + numItermax=numItermax, log=True) # check constratints np.testing.assert_allclose( |