From 9422dc98c2049a21ad845f7caac5093b89abcd6c Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Mon, 13 Apr 2026 13:29:00 +0200 Subject: [PATCH] fix: remove leftover normalization calls from reward classifier predict_reward Fixes #2355 --- src/lerobot/policies/sac/reward_model/modeling_classifier.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/lerobot/policies/sac/reward_model/modeling_classifier.py b/src/lerobot/policies/sac/reward_model/modeling_classifier.py index dba6a174b..5fb2abfde 100644 --- a/src/lerobot/policies/sac/reward_model/modeling_classifier.py +++ b/src/lerobot/policies/sac/reward_model/modeling_classifier.py @@ -268,10 +268,6 @@ class Classifier(PreTrainedPolicy): def predict_reward(self, batch, threshold=0.5): """Eval method. Returns predicted reward with the decision threshold as argument.""" - # Check for both OBS_IMAGE and OBS_IMAGES prefixes - batch = self.normalize_inputs(batch) - batch = self.normalize_targets(batch) - # Extract images from batch dict images = [batch[key] for key in self.config.input_features if key.startswith(OBS_IMAGE)]