From 547a03ef87e4aa92edc1e89ee2db04114e1a8ad5 Mon Sep 17 00:00:00 2001
From: ievred <ievgen.redko@univ-st-etienne.fr>
Date: Wed, 1 Apr 2020 09:13:58 +0200
Subject: fix test example add M to log

---
 ot/bregman.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'ot/bregman.py')

diff --git a/ot/bregman.py b/ot/bregman.py
index d17aaf0..fb959e9 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -1603,6 +1603,7 @@ def jcpot_barycenter(Xs, Ys, Xt, reg, metric='sqeuclidean', numItermax=100,
         # build the cost matrix and the Gibbs kernel
         M = dist(Xs[d], Xt, metric=metric)
         M = M / np.median(M)
+        dom['M'] = M
 
         K = np.empty(M.shape, dtype=M.dtype)
         np.divide(M, -reg, out=K)
-- 
cgit v1.2.3