From cfe914f730188685dc8580f5cdbb305b96e1e621 Mon Sep 17 00:00:00 2001 From: Jacan Chaplais Date: Thu, 23 Nov 2023 16:13:02 +0000 Subject: [PATCH] improved pt distance precision with expm1 #168 --- graphicle/calculate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphicle/calculate.py b/graphicle/calculate.py index 55a45c0..f769c1a 100644 --- a/graphicle/calculate.py +++ b/graphicle/calculate.py @@ -656,4 +656,4 @@ def aggregate_momenta( @nb.vectorize("float64(float64, float64)") def _pt_distance(pt_1: float, pt_2: float) -> float: - return 1.0 - math.exp(-0.5 * pow((pt_1 - pt_2) / min(pt_1, pt_2), 2)) + return -math.expm1(-0.5 * pow((pt_1 - pt_2) / min(pt_1, pt_2), 2))