From 81ebb6fe9423299f633e55071985123a517da3bd Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 10 Sep 2025 15:19:44 +0200 Subject: [PATCH] fix(reforcexy): integer type casting on None 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 | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 85c1e83..e0827a4 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -253,8 +253,9 @@ class ReforceXY(BaseReinforcementLearningModel): logger.info("Clip range linear schedule enabled, initial value: %s", cr) if "DQN" in self.model_type: - gradient_steps = int(model_params.get("gradient_steps")) + gradient_steps = model_params.get("gradient_steps") if gradient_steps is None: + gradient_steps = int(gradient_steps) train_freq = model_params.get("train_freq") if isinstance(train_freq, (tuple, list)) and train_freq: train_freq = ( @@ -273,7 +274,7 @@ class ReforceXY(BaseReinforcementLearningModel): train_freq, max(train_freq // subsample_steps, 1) ) else: - model_params["gradient_steps"] = 1 + model_params["gradient_steps"] = -1 if not model_params.get("policy_kwargs"): model_params["policy_kwargs"] = {} -- 2.43.0