From 7a672edc34301f28465ebe73d2e4c801f07db351 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 10 Sep 2025 21:12:52 +0200 Subject: [PATCH] fix(reforcexy): logic fix at step validation MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/user_data/freqaimodels/ReforceXY.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 4b96b03..3b1729a 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -1593,7 +1593,9 @@ class MaskableTrialEvalCallback(MaskableEvalCallback): if self.is_pruned: return False _super_on_step = super()._on_step() - if _super_on_step and self.eval_freq > 0 and self.n_calls % self.eval_freq == 0: + if not _super_on_step: + return False + if self.eval_freq > 0 and self.n_calls % self.eval_freq == 0: self.eval_idx += 1 last_mean_reward = getattr(self, "last_mean_reward", np.nan) if not isinstance(last_mean_reward, (int, float)) or not np.isfinite( -- 2.43.0