From 0138dcf636c3f3f0e63110b08a8249f065e1fa73 Mon Sep 17 00:00:00 2001
From: Nathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com>
Date: Mon, 1 Aug 2022 17:38:05 +0200
Subject: [MRG] Solve example throwing an error when executed on a GPU (#391)

* Solve example throwing an error when executed on a GPU

* add PR to releases.md

* update pep8 command

* pep8
---
 ot/gromov.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'ot/gromov.py')

diff --git a/ot/gromov.py b/ot/gromov.py
index 55ab0bd..7de8598 100644
--- a/ot/gromov.py
+++ b/ot/gromov.py
@@ -1581,7 +1581,7 @@ def gromov_barycenters(N, Cs, ps, p, lambdas, loss_fun,
 
     error = []
 
-    while(err > tol and cpt < max_iter):
+    while (err > tol and cpt < max_iter):
         Cprev = C
 
         T = [gromov_wasserstein(Cs[s], C, ps[s], p, loss_fun,
@@ -1725,7 +1725,7 @@ def fgw_barycenters(N, Ys, Cs, ps, lambdas, alpha, fixed_structure=False, fixed_
         log_['err_structure'] = []
         log_['Ts_iter'] = []
 
-    while((err_feature > tol or err_structure > tol) and cpt < max_iter):
+    while ((err_feature > tol or err_structure > tol) and cpt < max_iter):
         Cprev = C
         Xprev = X
 
-- 
cgit v1.2.3