diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-11-19 11:18:29 +0100 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-11-19 11:18:29 +0100 |
| commit | de04afc0f9f01fc09a3a8138865eacc0b6f4415d (patch) | |
| tree | 876662abd26be98d6b80af820792e993f0865b94 /ot/gpu/bregman.py | |
| parent | 93db239e1156ad1db8edbb13c1ecde973ce009c0 (diff) | |
update flake8 parameters
Diffstat (limited to 'ot/gpu/bregman.py')
| -rw-r--r-- | ot/gpu/bregman.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ot/gpu/bregman.py b/ot/gpu/bregman.py index 3031ed9..978b307 100644 --- a/ot/gpu/bregman.py +++ b/ot/gpu/bregman.py @@ -146,9 +146,9 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, stopThr=1e-9, v = np.divide(b, KtransposeU) u = 1. / np.dot(Kp, v) - if (np.any(KtransposeU == 0) - or np.any(np.isnan(u)) or np.any(np.isnan(v)) - or np.any(np.isinf(u)) or np.any(np.isinf(v))): + if (np.any(KtransposeU == 0) or + np.any(np.isnan(u)) or np.any(np.isnan(v)) or + np.any(np.isinf(u)) or np.any(np.isinf(v))): # we have reached the machine precision # come back to previous solution and quit loop print('Warning: numerical errors at iteration', cpt) |