[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

3 files changed, 5 insertions, 5 deletions

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

 

diff --git a/ot/lp/cvx.py b/ot/lp/cvx.py
index fbf3c0e..361ad0f 100644
--- a/ot/lp/cvx.py
+++ b/ot/lp/cvx.py
@@ -80,7 +80,7 @@ def barycenter(A, M, weights=None, verbose=False, log=False, solver='interior-po
     if weights is None:
         weights = np.ones(A.shape[1]) / A.shape[1]
     else:
-        assert(len(weights) == A.shape[1])
+        assert len(weights) == A.shape[1]
 
     n_distributions = A.shape[1]
     n = A.shape[0]
diff --git a/ot/unbalanced.py b/ot/unbalanced.py
index 90c920c..dd9a36e 100644
--- a/ot/unbalanced.py
+++ b/ot/unbalanced.py
@@ -734,7 +734,7 @@ def barycenter_unbalanced_stabilized(A, M, reg, reg_m, weights=None, tau=1e3,
     if weights is None:
         weights = nx.ones(n_hists, type_as=A) / n_hists
     else:
-        assert(len(weights) == A.shape[1])
+        assert len(weights) == A.shape[1]
 
     if log:
         log = {'err': []}
@@ -882,7 +882,7 @@ def barycenter_unbalanced_sinkhorn(A, M, reg, reg_m, weights=None,
     if weights is None:
         weights = nx.ones(n_hists, type_as=A) / n_hists
     else:
-        assert(len(weights) == A.shape[1])
+        assert len(weights) == A.shape[1]
 
     if log:
         log = {'err': []}