Skip to content

Commit 30fc233

Browse files
committed
correct pep8
1 parent 3844639 commit 30fc233

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

test/test_ot.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -337,10 +337,11 @@ def test_dual_variables():
337337
# Check that both cost computations are equivalent
338338
np.testing.assert_almost_equal(cost1, log['cost'])
339339
check_duality_gap(a, b, M, G, log['u'], log['v'], log['cost'])
340-
341-
viol=log['u'][:,None]+log['v'][None,:]-M
342-
343-
assert viol.max()<1e-8
340+
341+
viol = log['u'][:, None] + log['v'][None, :] - M
342+
343+
assert viol.max() < 1e-8
344+
344345

345346
def check_duality_gap(a, b, M, G, u, v, cost):
346347
cost_dual = np.vdot(a, u) + np.vdot(b, v)

0 commit comments

Comments
 (0)