fix bug in doctest

Author: 6Ulm

ot/unbalanced.py

@@ -110,9 +110,8 @@ def sinkhorn_unbalanced(a, b, M, reg, reg_m, method='sinkhorn',
     >>> b=[.5, .5]
     >>> M=[[0., 1.], [1., 0.]]
     >>> ot.sinkhorn_unbalanced(a, b, M, 1, 1)
-    array([[0.51122823, 0.18807035],
-           [0.18807035, 0.51122823]])
-
+    array([[0.51122814, 0.18807032],
+           [0.18807032, 0.51122814]])
 
     .. _references-sinkhorn-unbalanced:
     References
@@ -249,8 +248,7 @@ def sinkhorn_unbalanced2(a, b, M, reg, reg_m, method='sinkhorn',
     >>> b=[.5, .5]
     >>> M=[[0., 1.],[1., 0.]]
     >>> ot.unbalanced.sinkhorn_unbalanced2(a, b, M, 1., 1.)
-    array([0.31912866])
-
+    0.3191285827553562
 
     .. _references-sinkhorn-unbalanced2:
     References
@@ -415,9 +413,8 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, reg_m, reg_type="entropy",
     >>> b=[.5, .5]
     >>> M=[[0., 1.],[1., 0.]]
     >>> ot.unbalanced.sinkhorn_knopp_unbalanced(a, b, M, 1., 1.)
-    array([[0.51122823, 0.18807035],
-           [0.18807035, 0.51122823]])
-
+    array([[0.51122814, 0.18807032],
+           [0.18807032, 0.51122814]])
 
     .. _references-sinkhorn-knopp-unbalanced:
     References
@@ -625,9 +622,8 @@ def sinkhorn_stabilized_unbalanced(a, b, M, reg, reg_m, reg_type="entropy",
     >>> b=[.5, .5]
     >>> M=[[0., 1.],[1., 0.]]
     >>> ot.unbalanced.sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.)
-    array([[0.51122823, 0.18807035],
-           [0.18807035, 0.51122823]])
-
+    array([[0.51122814, 0.18807032],
+           [0.18807032, 0.51122814]])
 
     .. _references-sinkhorn-stabilized-unbalanced:
     References