From dcc0c234dd4c30b0a105cc5b5d8fad0c0821445e Mon Sep 17 00:00:00 2001
From: tidely <43219534+tidely@users.noreply.github.com>
Date: Sat, 14 Jun 2025 15:06:22 +0300
Subject: [PATCH] Improve type hints (#1293)

---
 lerobot/common/robots/robot.py               | 4 ++--
 lerobot/common/teleoperators/teleoperator.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

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):