Skip to content

Commit 6a31a6b

Browse files
committed
more bar iterations for auto ground bary test
1 parent bb521c3 commit 6a31a6b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/test_ot.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,7 @@ def ground_bary(y):
561561
cost_list,
562562
ground_bary,
563563
numItermax=1,
564+
stopThr=-1,
564565
)
565566

566567
X2, log2 = ot.lp.free_support_barycenter_generic_costs(
@@ -573,6 +574,7 @@ def ground_bary(y):
573574
ground_bary_stopThr=1e-20,
574575
ground_bary_numItermax=100,
575576
numItermax=10,
577+
stopThr=-1,
576578
log=True,
577579
)
578580

@@ -589,6 +591,7 @@ def ground_bary(y):
589591
ground_bary_numItermax=100,
590592
numItermax=10,
591593
ground_bary_solver="Adam",
594+
stopThr=-1,
592595
)
593596

594597
np.testing.assert_allclose(X2.numpy(), X3.numpy(), rtol=1e-3, atol=1e-3)

0 commit comments

Comments
 (0)