Skip to content

Commit 49d9b5c

Browse files
author
Hicham Janati
committed
fix doctest examples
1 parent 7efea81 commit 49d9b5c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ot/unbalanced.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, reg_m, numItermax=1000,
326326
>>> a=[.5, .5]
327327
>>> b=[.5, .5]
328328
>>> M=[[0., 1.],[1., 0.]]
329-
>>> ot.unbalanced._sinkhorn_knopp_unbalanced(a, b, M, 1., 1.)
329+
>>> ot.unbalanced.sinkhorn_knopp_unbalanced(a, b, M, 1., 1.)
330330
array([[0.51122823, 0.18807035],
331331
[0.18807035, 0.51122823]])
332332
@@ -510,7 +510,7 @@ def sinkhorn_stabilized_unbalanced(a, b, M, reg, reg_m, tau=1e5, numItermax=1000
510510
>>> a=[.5, .5]
511511
>>> b=[.5, .5]
512512
>>> M=[[0., 1.],[1., 0.]]
513-
>>> ot.unbalanced._sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.)
513+
>>> ot.unbalanced.sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.)
514514
array([[0.51122823, 0.18807035],
515515
[0.18807035, 0.51122823]])
516516

0 commit comments

Comments
 (0)