From d674a56b7a5f5b56c69e40b9e86d3bbe3f7ec742 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 7 Oct 2025 11:25:35 +0200 Subject: [PATCH] test(reforcexy): cleanup imports MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../reward_space_analysis/test_reward_space_analysis.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ReforceXY/reward_space_analysis/test_reward_space_analysis.py b/ReforceXY/reward_space_analysis/test_reward_space_analysis.py index 563bb8d..722f0ba 100644 --- a/ReforceXY/reward_space_analysis/test_reward_space_analysis.py +++ b/ReforceXY/reward_space_analysis/test_reward_space_analysis.py @@ -23,7 +23,6 @@ import pandas as pd # Import functions to test try: - import reward_space_analysis as rsa from reward_space_analysis import ( DEFAULT_MODEL_REWARD_PARAMETERS, Actions, @@ -33,6 +32,7 @@ try: bootstrap_confidence_intervals, calculate_reward, compute_distribution_shift_metrics, + compute_exit_factor, distribution_diagnostics, parse_overrides, simulate_samples, @@ -673,7 +673,7 @@ class TestRewardAlignment(RewardSpaceTestBase): for mode in modes_to_test: test_params = self.DEFAULT_PARAMS.copy() test_params["exit_attenuation_mode"] = mode - factor = rsa.compute_exit_factor( + factor = compute_exit_factor( base_factor=1.0, pnl=0.02, pnl_factor=1.5, @@ -695,14 +695,14 @@ class TestRewardAlignment(RewardSpaceTestBase): "exit_linear_slope": 1.0, } ) - plateau_factor_pre = rsa.compute_exit_factor( + plateau_factor_pre = compute_exit_factor( base_factor=1.0, pnl=0.02, pnl_factor=1.5, duration_ratio=0.4, # inside grace params=plateau_params, ) - plateau_factor_post = rsa.compute_exit_factor( + plateau_factor_post = compute_exit_factor( base_factor=1.0, pnl=0.02, pnl_factor=1.5, -- 2.53.0