From 2e736028d1cbb742f901757c5da94ed6182a8bb3 Mon Sep 17 00:00:00 2001
From: ferrari <maxence.ferrari@gmail.com>
Date: Thu, 4 May 2023 17:20:36 +0200
Subject: [PATCH] More aggressive fix floating point error
---
gsrp_smart_util.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/gsrp_smart_util.py b/gsrp_smart_util.py
index 0e42e8d..8dfb6a2 100644
--- a/gsrp_smart_util.py
+++ b/gsrp_smart_util.py
@@ -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)
--
GitLab