Fix: add kwargs in reward classifier __init__() · e6d282108d - lerobot - Gitea: Git with a cup of tea

admin/lerobot

mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00

Fix: add kwargs in reward classifier init()

This commit is contained in:

Khalil Meftah

2026-04-14 11:13:43 +02:00

parent a8838c081b

commit e6d282108d

1 changed files with 1 additions and 0 deletions

									
										src/lerobot/policies/sac/reward_model/modeling_classifier.py
									
		+1
		
												View File
												
					@@ -107,6 +107,7 @@ class Classifier(PreTrainedPolicy):

					    def __init__(

					    def __init__(

					        self,

					        self,

					        config: RewardClassifierConfig,

					        config: RewardClassifierConfig,

					        **kwargs,

					    ):

					    ):

					        from transformers import AutoModel

					        from transformers import AutoModel