From 178c281fc91e014f5e148b7017430928d715de8c Mon Sep 17 00:00:00 2001
From: ncassereau-idris <84033440+ncassereau-idris@users.noreply.github.com>
Date: Wed, 26 May 2021 16:56:00 +0200
Subject: Docs correction (#252)

---
 ot/lp/__init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'ot/lp/__init__.py')

diff --git a/ot/lp/__init__.py b/ot/lp/__init__.py
index f08e020..d5c3a5e 100644
--- a/ot/lp/__init__.py
+++ b/ot/lp/__init__.py
@@ -323,7 +323,7 @@ def emd2(a, b, M, processes=multiprocessing.cpu_count(),
         The maximum number of iterations before stopping the optimization
         algorithm if it has not converged.
     log: boolean, optional (default=False)
-        If True, returns a dictionary containing the cost and dual
+        If True, returns a dictionary containing dual
         variables. Otherwise returns only the optimal transportation cost.
     return_matrix: boolean, optional (default=False)
         If True, returns the optimal transportation matrix in the log.
@@ -333,10 +333,10 @@ def emd2(a, b, M, processes=multiprocessing.cpu_count(),
 
     Returns
     -------
-    gamma: (ns x nt) ndarray
-        Optimal transportation matrix for the given parameters
+    W: float
+        Optimal transportation loss for the given parameters
     log: dictnp
-        If input log is true, a dictionary containing the cost and dual
+        If input log is true, a dictionary containing dual
         variables and exit status
 
 
-- 
cgit v1.2.3