fix(reforcexy): logic fix at step validation

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index 4b96b03bae0c1454a3a10e53497900ab070f6f92..3b1729ab50bf1faa692eebcf3387ed2aa8532480 100644 (file)
--- 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(