diff options
Diffstat (limited to 'test/test_stochastic.py')
-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 f1d4825..10e62c1 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -196,8 +196,8 @@ def test_dual_sgd_sinkhorn(): reg = 1 batch_size = 30 - a = ot.datasets.get_1D_gauss(n, 15, 5) # m= mean, s= std - b = ot.datasets.get_1D_gauss(n, 15, 5) + a = ot.datasets.make_1D_gauss(n, 15, 5) # m= mean, s= std + b = ot.datasets.make_1D_gauss(n, 15, 5) X_source = np.arange(n, dtype=np.float64) Y_target = np.arange(n, dtype=np.float64) M = ot.dist(X_source.reshape((n, 1)), Y_target.reshape((n, 1))) |