Bugfix (python2 in unbalanced)

Author: rtavenar

ot/unbalanced.py

@@ -6,6 +6,7 @@
 # Author: Hicham Janati <[email protected]>
 # License: MIT License
 
+from __future__ import division
 import warnings
 import numpy as np
 # from .utils import unif, dist
@@ -287,12 +288,12 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, alpha, numItermax=1000,
     --------
 
     >>> import ot
-    >>> a=[.5, .15]
+    >>> a=[.5, .5]
     >>> b=[.5, .5]
     >>> M=[[0., 1.],[1., 0.]]
     >>> ot.unbalanced.sinkhorn_knopp_unbalanced(a, b, M, 1., 1.)
-    array([[0.52761554, 0.22392482],
-           [0.10286295, 0.32257641]])
+    array([[0.51122823, 0.18807035],
+           [0.18807035, 0.51122823]])
 
     References
     ----------