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/unbalanced.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/unbalanced.py')
| -rw-r--r-- | ot/unbalanced.py | 4 |
1 files changed, 2 insertions, 2 deletions
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': []} |