diff options
| author | Nathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com> | 2022-08-01 17:38:05 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-08-01 17:38:05 +0200 |
| commit | 0138dcf636c3f3f0e63110b08a8249f065e1fa73 (patch) | |
| tree | 704ed82dff4c5807ef983c64a60bad6a0e54ef3d /ot/lp/cvx.py | |
| parent | 818c7ace20da36d8042b0d7ad7a712b27f7afd59 (diff) | |
[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
Diffstat (limited to 'ot/lp/cvx.py')
| -rw-r--r-- | ot/lp/cvx.py | 2 |
1 files changed, 1 insertions, 1 deletions
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] |