hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Fix config parsing

Browse Source
This commit is contained in:
Simon Alibert
2025-05-08 12:59:15 +02:00
parent a0e69dd708
commit 73ac4f38b2
3 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/configs/parser.py
View File

@@ -26,7 +26,6 @@ from lerobot.common.utils.utils import has_method
PATH_KEY = "path" PATH_KEY = "path"
PLUGIN_DISCOVERY_SUFFIX = "discover_packages_path" PLUGIN_DISCOVERY_SUFFIX = "discover_packages_path"
draccus.set_config_type("json")
def get_cli_overrides(field_name: str, args: Sequence[str] | None = None) -> list[str] | None: def get_cli_overrides(field_name: str, args: Sequence[str] | None = None) -> list[str] | None:

3
lerobot/configs/policies.py
View File

@@ -173,4 +173,5 @@ class PreTrainedConfig(draccus.ChoiceRegistry, HubMixin, abc.ABC):
# HACK: this is very ugly, ideally we'd like to be able to do that natively with draccus # HACK: this is very ugly, ideally we'd like to be able to do that natively with draccus
# something like --policy.path (in addition to --policy.type) # something like --policy.path (in addition to --policy.type)
cli_overrides = policy_kwargs.pop("cli_overrides", []) cli_overrides = policy_kwargs.pop("cli_overrides", [])
return draccus.parse(cls, config_file, args=cli_overrides) with draccus.config_type("json"):
return draccus.parse(cls, config_file, args=cli_overrides)

5
lerobot/configs/train.py
View File

@@ -170,6 +170,5 @@ class TrainPipelineConfig(HubMixin):
) from e ) from e
cli_args = kwargs.pop("cli_args", []) cli_args = kwargs.pop("cli_args", [])
cfg = draccus.parse(cls, config_file, args=cli_args) with draccus.config_type("json"):
return draccus.parse(cls, config_file, args=cli_args)
return cfg