From 29106f8f272ca143ec091c67cd1e8faacc0725c8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Thu, 9 Oct 2025 22:24:19 +0200 Subject: [PATCH] test(reforcexy): fix np.isclose() invocation MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/reward_space_analysis/reward_space_analysis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ReforceXY/reward_space_analysis/reward_space_analysis.py b/ReforceXY/reward_space_analysis/reward_space_analysis.py index c9c88e0..eaf9b4f 100644 --- a/ReforceXY/reward_space_analysis/reward_space_analysis.py +++ b/ReforceXY/reward_space_analysis/reward_space_analysis.py @@ -1378,7 +1378,7 @@ def compute_distribution_shift_metrics( # Guard against degenerate distributions (all values identical) if not np.isfinite(min_val) or not np.isfinite(max_val): continue - if np.isclose(max_val, min_val, rel_tol=0, abs_tol=1e-12): + if np.isclose(max_val, min_val, rtol=0, atol=1e-12): # All mass at a single point -> shift metrics are all zero by definition metrics[f"{feature}_kl_divergence"] = 0.0 metrics[f"{feature}_js_distance"] = 0.0 -- 2.43.0