1 files changed, 2 insertions, 1 deletions
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