summaryrefslogtreecommitdiff
path: root/test/test_stochastic.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_stochastic.py')
-rw-r--r--test/test_stochastic.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py
index 8ddf485..155622c 100644
--- a/test/test_stochastic.py
+++ b/test/test_stochastic.py
@@ -71,7 +71,7 @@ def test_stochastic_asgd():
M = ot.dist(x, x)
G, log = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "asgd",
- numItermax=numItermax, log=True)
+ numItermax=numItermax, log=True)
# check constratints
np.testing.assert_allclose(
@@ -146,7 +146,7 @@ def test_stochastic_dual_sgd():
M = ot.dist(x, x)
G, log = ot.stochastic.solve_dual_entropic(u, u, M, reg, batch_size,
- numItermax=numItermax, log=True)
+ numItermax=numItermax, log=True)
# check constratints
np.testing.assert_allclose(