summaryrefslogtreecommitdiff
path: root/test/test_stochastic.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2020-04-24 12:39:38 +0200
committerRémi Flamary <remi.flamary@gmail.com>2020-04-24 12:39:38 +0200
commit53b063ed6b6aa15d6cb103a9304bbd169678b2e9 (patch)
tree33f75e733e1f93c07a5d37f72f085c9722bf19c7 /test/test_stochastic.py
parent46523dc0956fd17e709f958ebd351e748fca0a23 (diff)
better coverage options verbose and log
Diffstat (limited to 'test/test_stochastic.py')
-rw-r--r--test/test_stochastic.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py
index f0f3fc8..8ddf485 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(