summaryrefslogtreecommitdiff
path: root/ot/bregman.py
diff options
context:
space:
mode:
authorHicham Janati <hicham.janati@inria.fr>2019-06-18 16:40:06 +0200
committerHicham Janati <hicham.janati@inria.fr>2019-06-18 16:40:06 +0200
commit897982718a5fd81a9a591d80a7d50839399fc088 (patch)
tree48189493c09cda25ee19dfd0b7ef59c2f6819ba7 /ot/bregman.py
parent50bc90058940645a13e2f3e41129bdc97161dc63 (diff)
fix func names + add more tests
Diffstat (limited to 'ot/bregman.py')
-rw-r--r--ot/bregman.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman.py b/ot/bregman.py
index 321712b..09716e6 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -241,7 +241,7 @@ def sinkhorn2(a, b, M, reg, method='sinkhorn', numItermax=1000,
b = np.asarray(b, dtype=np.float64)
if len(b.shape) < 2:
- b = b.reshape((-1, 1))
+ b = b[:, None]
return sink()