From 9c5cc82b2e6f6dcd94a24a47f414c278c717ee7a Mon Sep 17 00:00:00 2001
From: Leo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr>
Date: Thu, 20 Apr 2017 14:06:17 +0200
Subject: missing argument function sinkhorn

---
 ot/gpu/da.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'ot/gpu/da.py')

diff --git a/ot/gpu/da.py b/ot/gpu/da.py
index 8b45772..4c583a7 100644
--- a/ot/gpu/da.py
+++ b/ot/gpu/da.py
@@ -77,5 +77,5 @@ class OTDA_sinkhorn_GPU(OTDA):
             M = np.log(1 + np.log(1 + self.M_GPU.asarray()))
             self.M_GPU = cudamat.CUDAMatrix(M)
 
-        self.G = sinkhorn(ws, wt, self.M_GPU, reg, cudamat=cudamat)
+        self.G = sinkhorn(ws, wt, self.M_GPU, reg, cudamat=cudamat, **kwargs)
         self.computed = True
-- 
cgit v1.2.3