From 75492827c89a47cbc6807d4859be178d255c49bc Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Mon, 24 Jul 2017 12:09:15 +0200
Subject: add test sinkhorn

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

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

diff --git a/ot/gpu/bregman.py b/ot/gpu/bregman.py
index 7881c65..2302f80 100644
--- a/ot/gpu/bregman.py
+++ b/ot/gpu/bregman.py
@@ -9,7 +9,7 @@ import cudamat
 
 def sinkhorn(a, b, M_GPU, reg, numItermax=1000, stopThr=1e-9, verbose=False,
                 log=False, returnAsGPU=False):
-    """
+    r"""
     Solve the entropic regularization optimal transport problem on GPU
 
     The function solves the following optimization problem:
-- 
cgit v1.2.3