diff options
| author | Gard Spreemann <gspr@nonempty.org> | 2020-07-09 08:49:39 +0200 |
|---|---|---|
| committer | Gard Spreemann <gspr@nonempty.org> | 2020-07-09 08:49:39 +0200 |
| commit | a16b9471d7114ec08977479b7249efe747702b97 (patch) | |
| tree | 692f3061d0329cad954206b2bf903ba3384403f0 /test/test_stochastic.py | |
| parent | 0812fcd82cbf11d444619e96c55ba507bc09ef5d (diff) | |
| parent | 94d5c8cc9046854f473d8e4526a3bcf214eb5411 (diff) | |
Merge tag '0.7.0' into dfsg/latest
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( |