From 80a6cee699e9170f439c906554c13d5e31ab3d83 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Sat, 8 Mar 2025 11:09:21 +0100 Subject: [PATCH] Fix typing --- lerobot/common/policies/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/policies/config.py b/lerobot/common/policies/config.py index 7caac957e..167b2ab7b 100644 --- a/lerobot/common/policies/config.py +++ b/lerobot/common/policies/config.py @@ -178,7 +178,7 @@ class PreTrainedConfig( return draccus.parse(cls, config_file, args=cli_overrides) @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.""" if config_type in cls._choice_registry and not exist_ok: raise ValueError(