From 075f5cd18e0f254cf233f67a9cb72d8bff41aab4 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Mon, 15 Sep 2025 02:53:21 +0200 Subject: [PATCH] refactor(reforcexy): remove uneeded check at frame stacking 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 | 8 -------- 1 file changed, 8 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 838ae51..12c4f3f 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -539,14 +539,6 @@ class ReforceXY(BaseReinforcementLearningModel): action_masks_param = {"action_masks": _action_masks(position)} np_observation = observation.to_numpy(dtype=np.float32) - shape = getattr(self, "shape", None) - if shape and np_observation.shape != shape: - logger.error( - "Frame shape mismatch: got %s expected %s", - np_observation.shape, - shape, - ) - raise ValueError("Frame shape mismatch") frame_stacking = self.frame_stacking if frame_stacking and frame_stacking > 1: -- 2.43.0