Skip to content

Commit de04afc

Browse files
committed
update flake8 parameters
1 parent 93db239 commit de04afc

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

ot/gpu/bregman.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -146,9 +146,9 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, stopThr=1e-9,
146146
v = np.divide(b, KtransposeU)
147147
u = 1. / np.dot(Kp, v)
148148

149-
if (np.any(KtransposeU == 0)
150-
or np.any(np.isnan(u)) or np.any(np.isnan(v))
151-
or np.any(np.isinf(u)) or np.any(np.isinf(v))):
149+
if (np.any(KtransposeU == 0) or
150+
np.any(np.isnan(u)) or np.any(np.isnan(v)) or
151+
np.any(np.isinf(u)) or np.any(np.isinf(v))):
152152
# we have reached the machine precision
153153
# come back to previous solution and quit loop
154154
print('Warning: numerical errors at iteration', cpt)

ot/stochastic.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -418,8 +418,8 @@ def solve_semi_dual_entropic(a, b, M, reg, method, numItermax=10000, lr=None,
418418
return None
419419

420420
opt_alpha = c_transform_entropic(b, M, reg, opt_beta)
421-
pi = (np.exp((opt_alpha[:, None] + opt_beta[None, :] - M[:, :]) / reg)
422-
* a[:, None] * b[None, :])
421+
pi = (np.exp((opt_alpha[:, None] + opt_beta[None, :] - M[:, :]) / reg) *
422+
a[:, None] * b[None, :])
423423

424424
if log:
425425
log = {}
@@ -520,8 +520,8 @@ def batch_grad_dual(a, b, M, reg, alpha, beta, batch_size, batch_alpha,
520520
arXiv preprint arxiv:1711.02283.
521521
'''
522522

523-
G = - (np.exp((alpha[batch_alpha, None] + beta[None, batch_beta]
524-
- M[batch_alpha, :][:, batch_beta]) / reg) *
523+
G = - (np.exp((alpha[batch_alpha, None] + beta[None, batch_beta] -
524+
M[batch_alpha, :][:, batch_beta]) / reg) *
525525
a[batch_alpha, None] * b[None, batch_beta])
526526
grad_beta = np.zeros(np.shape(M)[1])
527527
grad_alpha = np.zeros(np.shape(M)[0])
@@ -702,8 +702,8 @@ def solve_dual_entropic(a, b, M, reg, batch_size, numItermax=10000, lr=1,
702702

703703
opt_alpha, opt_beta = sgd_entropic_regularization(a, b, M, reg, batch_size,
704704
numItermax, lr)
705-
pi = (np.exp((opt_alpha[:, None] + opt_beta[None, :] - M[:, :]) / reg)
706-
* a[:, None] * b[None, :])
705+
pi = (np.exp((opt_alpha[:, None] + opt_beta[None, :] - M[:, :]) / reg) *
706+
a[:, None] * b[None, :])
707707
if log:
708708
log = {}
709709
log['alpha'] = opt_alpha

setup.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@ description-file = README.md
33

44
[flake8]
55
exclude = __init__.py
6-
ignore = E265,E501,W605
6+
ignore = E265,E501,W605,W503,W504

0 commit comments

Comments
 (0)