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

This commit is contained in:
Khalil Meftah
2026-04-21 14:07:14 +02:00
parent bbe974eb1e
commit 19697adf7a
+2 -2
View File
@@ -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