tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix typing

Browse Source
This commit is contained in:
Simon Alibert
2025-03-08 11:09:21 +01:00
parent 80387285bb
commit 80a6cee699
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/config.py
View File

@@ -178,7 +178,7 @@ class PreTrainedConfig(
return draccus.parse(cls, config_file, args=cli_overrides) return draccus.parse(cls, config_file, args=cli_overrides)
@classmethod @classmethod
def register(cls, config_type: str, config: "PreTrainedConfig", exist_ok: bool = False): def register(cls, config_type: str, config: Type[T], exist_ok: bool = False):
"""Register a new configuration for this class.""" """Register a new configuration for this class."""
if config_type in cls._choice_registry and not exist_ok: if config_type in cls._choice_registry and not exist_ok:
raise ValueError( raise ValueError(