From e6d282108d40b180ae495c6b5dbe331f19864987 Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Tue, 14 Apr 2026 11:13:43 +0200 Subject: [PATCH] Fix: add kwargs in reward classifier __init__() --- src/lerobot/policies/sac/reward_model/modeling_classifier.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/lerobot/policies/sac/reward_model/modeling_classifier.py b/src/lerobot/policies/sac/reward_model/modeling_classifier.py index 5fb2abfde..0d18d23d1 100644 --- a/src/lerobot/policies/sac/reward_model/modeling_classifier.py +++ b/src/lerobot/policies/sac/reward_model/modeling_classifier.py @@ -107,6 +107,7 @@ class Classifier(PreTrainedPolicy): def __init__( self, config: RewardClassifierConfig, + **kwargs, ): from transformers import AutoModel