Skip to content

Commit 50a5a41

Browse files
author
Hicham Janati
committed
fix doctest examples
1 parent 5c0ed10 commit 50a5a41

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

ot/unbalanced.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@ def sinkhorn_unbalanced(a, b, M, reg, mu, method='sinkhorn', numItermax=1000,
7777
>>> b=[.5, .5]
7878
>>> M=[[0., 1.], [1., 0.]]
7979
>>> ot.sinkhorn_unbalanced(a, b, M, 1, 1)
80-
array([[0.51122823, 0.18807035],
81-
[0.18807035, 0.51122823]])
80+
array([[0.51122818, 0.18807034],
81+
[0.18807034, 0.51122818]])
8282
8383
8484
References
@@ -193,7 +193,7 @@ def sinkhorn_unbalanced2(a, b, M, reg, mu, method='sinkhorn',
193193
>>> b=[.5, .5]
194194
>>> M=[[0., 1.],[1., 0.]]
195195
>>> ot.unbalanced.sinkhorn_unbalanced2(a, b, M, 1., 1.)
196-
array([0.31912866])
196+
array([0.31912862])
197197
198198
199199
@@ -308,8 +308,8 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, mu, numItermax=1000,
308308
>>> b=[.5, .5]
309309
>>> M=[[0., 1.],[1., 0.]]
310310
>>> ot.unbalanced.sinkhorn_knopp_unbalanced(a, b, M, 1., 1.)
311-
array([[0.51122823, 0.18807035],
312-
[0.18807035, 0.51122823]])
311+
array([[0.51122818, 0.18807034],
312+
[0.18807034, 0.51122818]])
313313
314314
References
315315
----------
@@ -479,8 +479,8 @@ def sinkhorn_stabilized_unbalanced(a, b, M, reg, mu, tau=1e5, numItermax=1000,
479479
>>> b=[.5, .5]
480480
>>> M=[[0., 1.],[1., 0.]]
481481
>>> ot.unbalanced.sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.)
482-
array([[0.51122823, 0.18807035],
483-
[0.18807035, 0.51122823]])
482+
array([[0.51122818, 0.18807034],
483+
[0.18807034, 0.51122818]])
484484
485485
References
486486
----------

0 commit comments

Comments
 (0)