diff --git a/lerobot/common/robots/robot.py b/lerobot/common/robots/robot.py index ec2b155f..76c57faf 100644 --- a/lerobot/common/robots/robot.py +++ b/lerobot/common/robots/robot.py @@ -14,7 +14,7 @@ import abc from pathlib import Path -from typing import Any +from typing import Any, Type import draccus @@ -39,7 +39,7 @@ class Robot(abc.ABC): """ # Set these in ALL subclasses - config_class: RobotConfig + config_class: Type[RobotConfig] name: str def __init__(self, config: RobotConfig): diff --git a/lerobot/common/teleoperators/teleoperator.py b/lerobot/common/teleoperators/teleoperator.py index a3851731..6a20a3a8 100644 --- a/lerobot/common/teleoperators/teleoperator.py +++ b/lerobot/common/teleoperators/teleoperator.py @@ -14,7 +14,7 @@ import abc from pathlib import Path -from typing import Any +from typing import Any, Type import draccus @@ -37,7 +37,7 @@ class Teleoperator(abc.ABC): """ # Set these in ALL subclasses - config_class: TeleoperatorConfig + config_class: Type[TeleoperatorConfig] name: str def __init__(self, config: TeleoperatorConfig):