Skip to content

Commit ea6642c

Browse files
fix failing test - cur_idx needs to be incremented by 1 after the loop
1 parent ea2890a commit ea6642c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ot/lp/emd_wrap.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,4 +203,5 @@ def emd_1d_sorted(np.ndarray[double, ndim=1, mode="c"] u_weights,
203203
w_i -= w_j
204204
w_j = v_weights[j]
205205
cur_idx += 1
206+
cur_idx += 1
206207
return G[:cur_idx], indices[:cur_idx], cost

0 commit comments

Comments
 (0)