diff options
| author | Kilian Fatras <kilianfatras@Kilians-MacBook-Air.local> | 2019-03-29 13:06:01 +0100 |
|---|---|---|
| committer | Kilian Fatras <kilianfatras@Kilians-MacBook-Air.local> | 2019-03-29 13:06:01 +0100 |
| commit | 9569f893defa8e712a4f3199770a0df745d4cfff (patch) | |
| tree | 2681de7cefc0c079aabaea65f1ff9c45bc495402 /test/test_bregman.py | |
| parent | a2545b5a503c95c9bf07948929b77e9c3f4f28d3 (diff) | |
fix pep8
Diffstat (limited to 'test/test_bregman.py')
| -rw-r--r-- | test/test_bregman.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index b890df1..8b001a7 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -195,13 +195,11 @@ def test_empirical_sinkhorn(): n = 100 a = ot.unif(n) b = ot.unif(n) - M = ot.dist(X_s, X_t) - M_e = ot.dist(X_s, X_t, metric='euclidean') - - rng = np.random.RandomState(0) X_s = np.reshape(np.arange(n), (n, 1)) X_t = np.reshape(np.arange(0, n), (n, 1)) + M = ot.dist(X_s, X_t) + M_e = ot.dist(X_s, X_t, metric='euclidean') G_sqe = ot.bregman.empirical_sinkhorn(X_s, X_t, 1) sinkhorn_sqe = ot.sinkhorn(a, b, M, 1) |