Skip to content

Commit ce86d14

Browse files
committed
Fix in test: no margin constraints here
1 parent 2633116 commit ce86d14

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

test/test_da.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -267,14 +267,6 @@ def test_unbalanced_sinkhorn_transport_class():
267267
assert_equal(otda.cost_.shape, ((Xs.shape[0], Xt.shape[0])))
268268
assert_equal(otda.coupling_.shape, ((Xs.shape[0], Xt.shape[0])))
269269

270-
# test margin constraints
271-
mu_s = unif(ns)
272-
mu_t = unif(nt)
273-
assert_allclose(
274-
np.sum(otda.coupling_, axis=0), mu_t, rtol=1e-3, atol=1e-3)
275-
assert_allclose(
276-
np.sum(otda.coupling_, axis=1), mu_s, rtol=1e-3, atol=1e-3)
277-
278270
# test transform
279271
transp_Xs = otda.transform(Xs=Xs)
280272
assert_equal(transp_Xs.shape, Xs.shape)

0 commit comments

Comments
 (0)