From 402f7542091fd8b4d98521e5b50dfef11c035061 Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Tue, 21 Apr 2026 14:11:30 +0200 Subject: [PATCH] refactor(classifier): remove redundant input normalization in predict_reward method --- src/lerobot/rewards/classifier/modeling_classifier.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/lerobot/rewards/classifier/modeling_classifier.py b/src/lerobot/rewards/classifier/modeling_classifier.py index 9b9815dd6..bedfffbe9 100644 --- a/src/lerobot/rewards/classifier/modeling_classifier.py +++ b/src/lerobot/rewards/classifier/modeling_classifier.py @@ -276,10 +276,6 @@ class Classifier(PreTrainedRewardModel): 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)]