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
lerobot/common/teleoperators/koch_leader/koch_leader.py
View File

@@ -69,12 +69,12 @@ class KochLeader(Teleoperator):
def is_connected(self) -> bool:
return self.arm.is_connected
def connect(self) -> None:
def connect(self, calibrate: bool = True) -> None:
if self.is_connected:
raise DeviceAlreadyConnectedError(f"{self} already connected")
self.arm.connect()
if not self.is_calibrated:
if not self.is_calibrated and calibrate:
self.calibrate()
self.configure()

4
lerobot/common/teleoperators/so100_leader/so100_leader.py
View File

@@ -66,12 +66,12 @@ class SO100Leader(Teleoperator):
def is_connected(self) -> bool:
return self.arm.is_connected
def connect(self) -> None:
def connect(self, calibrate: bool = True) -> None:
if self.is_connected:
raise DeviceAlreadyConnectedError(f"{self} already connected")
self.arm.connect()
if not self.is_calibrated:
if not self.is_calibrated and calibrate:
self.calibrate()
self.configure()

2
lerobot/common/teleoperators/teleoperator.py
View File

@@ -46,7 +46,7 @@ class Teleoperator(abc.ABC):
pass
@abc.abstractmethod
def connect(self) -> None:
def connect(self, calibrate: bool = True) -> None:
"""Connects to the teleoperator."""
pass

4
lerobot/common/teleoperators/widowx/widowx.py
View File

@@ -69,12 +69,12 @@ class WidowX(Teleoperator):
def is_connected(self) -> bool:
return self.arm.is_connected
def connect(self):
def connect(self, calibrate: bool = True):
if self.is_connected:
raise DeviceAlreadyConnectedError(f"{self} already connected")
self.arm.connect()
if not self.is_calibrated:
if not self.is_calibrated and calibrate:
self.calibrate()
self.configure()