From d370f18aaac4ed6903e3b4251c8b9c4685c53cc3 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Thu, 31 May 2018 13:40:45 +0200
Subject: bug verbose semi-dual

---
 ot/smooth.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'ot')

diff --git a/ot/smooth.py b/ot/smooth.py
index 4c46e73..dd734b3 100644
--- a/ot/smooth.py
+++ b/ot/smooth.py
@@ -580,10 +580,11 @@ def smooth_ot_semi_dual(a, b, M, reg, reg_type='l2', method="L-BFGS-B", stopThr=
         raise NotImplementedError('Unknown regularization')
 
     # solve dual
-    alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, tol=stopThr)
+    alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, 
+                                 tol=stopThr, verbose=verbose)
 
     # reconstruct transport matrix
-    G = get_plan_from_semi_dual(alpha, b, M, regul, verbose=verbose)
+    G = get_plan_from_semi_dual(alpha, b, M, regul)
 
     if log:
         log = {'alpha': alpha, 'res': res}
-- 
cgit v1.2.3