From 19697adf7a0b580b0cb5b4516f545e927f1956df Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Tue, 21 Apr 2026 14:07:14 +0200 Subject: [PATCH] refactor(train): update __get_path_fields__ to include reward_model for config loading --- src/lerobot/configs/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/configs/train.py b/src/lerobot/configs/train.py index f51cc211c..5516cf8c8 100644 --- a/src/lerobot/configs/train.py +++ b/src/lerobot/configs/train.py @@ -164,8 +164,8 @@ class TrainPipelineConfig(HubMixin): @classmethod def __get_path_fields__(cls) -> list[str]: - """This enables the parser to load config from the policy using `--policy.path=local/dir`""" - return ["policy"] + """Keys for draccus pretrained-path loading.""" + return ["policy", "reward_model"] 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