diff options
| author | Kilian Fatras <kilianfatras@dhcp-206-12-53-210.eduroam.wireless.ubc.ca> | 2018-06-18 18:05:48 -0700 |
|---|---|---|
| committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-210.eduroam.wireless.ubc.ca> | 2018-06-18 18:05:48 -0700 |
| commit | e068b58ba4234792d96287afd34c3cddef544dd4 (patch) | |
| tree | 9e03c736340fb0fab95a3880060044e58d1aa001 /test | |
| parent | 74cfe5ac77c3e964a85ef90c11d8ebffa16ddcfe (diff) | |
pep8
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_stochastic.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py index bc0cebb..5824df1 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -149,7 +149,7 @@ def test_stochastic_dual_sgd(): M = ot.dist(x, x) G = ot.stochastic.solve_dual_entropic(u, u, M, reg, batch_size, - numItermax=numItermax) + numItermax=numItermax) # check constratints np.testing.assert_allclose( @@ -180,7 +180,7 @@ def test_dual_sgd_sinkhorn(): M = ot.dist(x, x) G_sgd = ot.stochastic.solve_dual_entropic(u, u, M, reg, batch_size, - numItermax=nb_iter) + numItermax=nb_iter) G_sinkhorn = ot.sinkhorn(u, u, M, reg) |