From 613417653504f5ebd1bdc75200a1fe14ba6007a0 Mon Sep 17 00:00:00 2001 From: Wang Xinyan Date: Mon, 23 Oct 2023 00:57:32 +0800 Subject: [PATCH] Update qeq.py --- dmff/admp/qeq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dmff/admp/qeq.py b/dmff/admp/qeq.py index 1bd0fcb49..786f1e8db 100644 --- a/dmff/admp/qeq.py +++ b/dmff/admp/qeq.py @@ -69,7 +69,7 @@ def E_constP(q, lagmt, const_list, const_vals): @jit_condition() def mask_to_zero(v, mask): return jnp.piecewise( - v, [mask < 1e-5, mask >= 1e-5], [lambda x: CONST_0, lambda x: v] + v, [mask < 1e-4, mask >= 1e-4], [lambda x: CONST_0, lambda x: v] )