diff --git a/src/lerobot/record.py b/src/lerobot/record.py index d662efcab..904a1c7d0 100644 --- a/src/lerobot/record.py +++ b/src/lerobot/record.py @@ -83,6 +83,7 @@ from lerobot.robots import ( # noqa: F401 hope_jr, koch_follower, make_robot_from_config, + reachy2, so100_follower, so101_follower, ) @@ -93,6 +94,7 @@ from lerobot.teleoperators import ( # noqa: F401 homunculus, koch_leader, make_teleoperator_from_config, + reachy2_fake_teleoperator, so100_leader, so101_leader, ) diff --git a/src/lerobot/replay.py b/src/lerobot/replay.py index afe54d90f..2c1eafea0 100644 --- a/src/lerobot/replay.py +++ b/src/lerobot/replay.py @@ -55,6 +55,7 @@ from lerobot.robots import ( # noqa: F401 hope_jr, koch_follower, make_robot_from_config, + reachy2, so100_follower, so101_follower, ) diff --git a/src/lerobot/teleoperators/utils.py b/src/lerobot/teleoperators/utils.py index 344a95d72..53bd7bb08 100644 --- a/src/lerobot/teleoperators/utils.py +++ b/src/lerobot/teleoperators/utils.py @@ -65,5 +65,9 @@ def make_teleoperator_from_config(config: TeleoperatorConfig) -> Teleoperator: from .bi_so100_leader import BiSO100Leader return BiSO100Leader(config) + elif config.type == "reachy2_fake_teleoperator": + from .reachy2_fake_teleoperator import Reachy2FakeTeleoperator + + return Reachy2FakeTeleoperator(config) else: raise ValueError(config.type)