Fix config parsing

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

View File

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