admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-26 11:16:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(train): update __get_path_fields__ to include reward_model for config loading

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-21 14:07:14 +02:00
parent bbe974eb1e
commit 19697adf7a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/configs/train.py
+2 -2
View File
@@ -164,8 +164,8 @@ class TrainPipelineConfig(HubMixin):
@classmethod @classmethod
def __get_path_fields__(cls) -> list[str]: def __get_path_fields__(cls) -> list[str]:
"""This enables the parser to load config from the policy using `--policy.path=local/dir`""" """Keys for draccus pretrained-path loading."""
return ["policy"] return ["policy", "reward_model"]
def to_dict(self) -> dict[str, Any]: def to_dict(self) -> dict[str, Any]:
return draccus.encode(self) # type: ignore[no-any-return] # because of the third-party library draccus uses Any as the return type return draccus.encode(self) # type: ignore[no-any-return] # because of the third-party library draccus uses Any as the return type