Skip to content

Commit aad4169

Browse files
committed
Fix random_state for SinkhornL1l2Transport test
1 parent 3478d0a commit aad4169

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/test_da.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -159,12 +159,12 @@ def test_sinkhorn_l1l2_transport_class(nx):
159159
ns = 50
160160
nt = 50
161161

162-
Xs, ys = make_data_classif('3gauss', ns)
163-
Xt, yt = make_data_classif('3gauss2', nt)
162+
Xs, ys = make_data_classif('3gauss', ns, random_state=42)
163+
Xt, yt = make_data_classif('3gauss2', nt, random_state=43)
164164

165165
Xs, ys, Xt, yt = nx.from_numpy(Xs, ys, Xt, yt)
166166

167-
otda = ot.da.SinkhornL1l2Transport()
167+
otda = ot.da.SinkhornL1l2Transport(max_inner_iter=500)
168168

169169
# test its computed
170170
with warnings.catch_warnings():

0 commit comments

Comments
 (0)