fix: remove leftover normalization calls from reward classifier predict_reward

Fixes #2355
This commit is contained in:
Khalil Meftah
2026-04-13 13:29:00 +02:00
parent 11a0b0174f
commit 9422dc98c2
@@ -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)]