From b3fb1ef40a482f0989686b79373060d764b62d38 Mon Sep 17 00:00:00 2001
From: "Mokhtar Z. Alaya" <mzalaya@Mokhtars-iMac.local>
Date: Sat, 18 Jan 2020 07:45:34 +0100
Subject: clean

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

(limited to 'ot/bregman.py')

diff --git a/ot/bregman.py b/ot/bregman.py
index aff9f8c..c304b5d 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -2117,10 +2117,11 @@ def screenkhorn(a, b, M, reg, ns_budget=None, nt_budget=None, uniform=False, res
         log['v'] = vsc_full
         log['Isel'] = Isel
         log['Jsel'] = Jsel
+
     gamma = usc_full[:, None] * K * vsc_full[None, :]
     gamma = gamma / gamma.sum()
 
     if log:
         return gamma, log
     else:
-        return gamma
\ No newline at end of file
+        return gamma
-- 
cgit v1.2.3