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