diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-11-19 11:17:07 +0100 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-11-19 11:17:07 +0100 |
| commit | 93db239e1156ad1db8edbb13c1ecde973ce009c0 (patch) | |
| tree | 14101caa2699d0b90d165303658f1540d1e87a5c /ot/gpu/bregman.py | |
| parent | 87930c4bcddfded480983343ecc68c6b94bcce14 (diff) | |
remove W605 errors
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 978b307..3031ed9 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) |