]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
fix(reforcexy): move attribute init to proper code path
authorJérôme Benoit <jerome.benoit@sap.com>
Mon, 3 Mar 2025 18:15:20 +0000 (19:15 +0100)
committerJérôme Benoit <jerome.benoit@sap.com>
Mon, 3 Mar 2025 18:15:20 +0000 (19:15 +0100)
Signed-off-by: Jérôme Benoit <jerome.benoit@sap.com>
ReforceXY/user_data/freqaimodels/ReforceXY.py

index c6b3b3b54e25304050e8805f8a2a4633a4993c4f..7274b3a69965c5732a77c766569714f6872246cd 100644 (file)
@@ -682,9 +682,6 @@ class ReforceXY(BaseReinforcementLearningModel):
             self.timeout: int = self.rl_config.get("max_trade_duration_candles", 128)
             self._last_closed_position: Positions = None
             self._last_closed_trade_tick: int = 0
-            if self.add_state_info:
-                # STATE_INFO
-                self.state_features = ["pnl", "position", "trade_duration"]
             if self.force_actions:
                 logger.info(
                     "%s - take_profit: %s, stop_loss: %s, timeout: %s candles (%s days), observation_space: %s",
@@ -709,6 +706,8 @@ class ReforceXY(BaseReinforcementLearningModel):
             """
             super().reset_env(df, prices, window_size, reward_kwargs, starting_point)
             if self.add_state_info:
+                # STATE_INFO
+                self.state_features = ["pnl", "position", "trade_duration"]
                 self.total_features = self.signal_features.shape[1] + len(
                     self.state_features
                 )