We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bb521c3 commit 6a31a6bCopy full SHA for 6a31a6b
test/test_ot.py
@@ -561,6 +561,7 @@ def ground_bary(y):
561
cost_list,
562
ground_bary,
563
numItermax=1,
564
+ stopThr=-1,
565
)
566
567
X2, log2 = ot.lp.free_support_barycenter_generic_costs(
@@ -573,6 +574,7 @@ def ground_bary(y):
573
574
ground_bary_stopThr=1e-20,
575
ground_bary_numItermax=100,
576
numItermax=10,
577
578
log=True,
579
580
@@ -589,6 +591,7 @@ def ground_bary(y):
589
591
590
592
593
ground_bary_solver="Adam",
594
595
596
597
np.testing.assert_allclose(X2.numpy(), X3.numpy(), rtol=1e-3, atol=1e-3)
0 commit comments