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/unbalanced.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'ot/unbalanced.py')

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': []}
-- 
cgit v1.2.3