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

Add calibrate

Browse Source
This commit is contained in:
Simon Alibert
2025-05-08 18:27:19 +02:00
parent dd3e305164
commit 64303781c2
14 changed files with 91 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/mocks/mock_robot.py
View File

@@ -67,11 +67,13 @@ class MockRobot(Robot):
def is_connected(self) -> bool:
return self._is_connected
def connect(self) -> None:
def connect(self, calibrate: bool = True) -> None:
if self.is_connected:
raise DeviceAlreadyConnectedError(f"{self} already connected")
self._is_connected = True
if calibrate:
self.calibrate()
@property
def is_calibrated(self) -> bool:

4
tests/mocks/mock_teleop.py
View File

@@ -51,11 +51,13 @@ class MockTeleop(Teleoperator):
def is_connected(self) -> bool:
return self._is_connected
def connect(self) -> None:
def connect(self, calibrate: bool = True) -> None:
if self.is_connected:
raise DeviceAlreadyConnectedError(f"{self} already connected")
self._is_connected = True
if calibrate:
self.calibrate()
@property
def is_calibrated(self) -> bool: