Skip to content
Snippets Groups Projects
Commit 2e736028 authored by ferrari's avatar ferrari
Browse files

More aggressive fix floating point error

parent 32f67aa7
Branches
No related tags found
No related merge requests found
......@@ -127,8 +127,8 @@ def add(mem1, mem2, cc, t_max, id1, id2, n_ind, mem_limit=np.infty):
return out_val, out_tij
def mask_val(mem, val):
mask = mem[0] >= val
def mask_val(mem, val, atol=5e-9):
mask = mem[0] >= val - atol
return mem[0][mask], np.compress(mask, mem[1], axis=1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment