]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
fix(reforcexy): fix stacked observations shape
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Mon, 15 Sep 2025 00:15:24 +0000 (02:15 +0200)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Mon, 15 Sep 2025 00:15:24 +0000 (02:15 +0200)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
ReforceXY/user_data/freqaimodels/ReforceXY.py

index a6deefce0ad0b5a8e220f5f5e3584ae67a5ba8fa..b9efa731bd36b8d31fff6df4cbc3a47818207cfb 100644 (file)
@@ -558,6 +558,11 @@ class ReforceXY(BaseReinforcementLearningModel):
             else:
                 observations = np_observation.flatten()
 
+            if observations.ndim == 1:
+                observations = observations.reshape(1, -1)
+            else:
+                observations = observations
+
             action, _ = model.predict(
                 observations, deterministic=True, **action_masks_param
             )