From 94eb4a24fe09c8a193933d8c7f48a657da4e6c52 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Fri, 11 May 2018 16:07:03 +0200
Subject: update documentation in bregman

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

(limited to 'ot/bregman.py')

diff --git a/ot/bregman.py b/ot/bregman.py
index 07b8660..9c84aed 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -844,6 +844,8 @@ def barycenter(A, M, reg, weights=None, numItermax=1000,
         loss matrix   for OT
     reg : float
         Regularization term >0
+    weights : np.ndarray (n,)
+        Weights of each histogram i_i on the simplex        
     numItermax : int, optional
         Max number of iterations
     stopThr : float, optional
-- 
cgit v1.2.3