Rename viperx & widowx

This commit is contained in:
Simon Alibert 2025-04-03 18:37:21 +02:00
parent f7672e14c7
commit 3848a8f9aa
6 changed files with 14 additions and 14 deletions

View File

@ -0,0 +1,2 @@
from .config_viperx import ViperXConfig
from .viperx import ViperX

View File

@ -7,7 +7,7 @@ from ..config import RobotConfig
@RobotConfig.register_subclass("viperx") @RobotConfig.register_subclass("viperx")
@dataclass @dataclass
class ViperXRobotConfig(RobotConfig): class ViperXConfig(RobotConfig):
port: str # Port to connect to the arm port: str # Port to connect to the arm
disable_torque_on_disconnect: bool = True disable_torque_on_disconnect: bool = True

View File

@ -19,22 +19,22 @@ from lerobot.common.motors.dynamixel import (
from ..robot import Robot from ..robot import Robot
from ..utils import ensure_safe_goal_position from ..utils import ensure_safe_goal_position
from .configuration_viperx import ViperXRobotConfig from .config_viperx import ViperXConfig
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class ViperXRobot(Robot): class ViperX(Robot):
""" """
[ViperX](https://www.trossenrobotics.com/viperx-300) developed by Trossen Robotics [ViperX](https://www.trossenrobotics.com/viperx-300) developed by Trossen Robotics
""" """
config_class = ViperXRobotConfig config_class = ViperXConfig
name = "viperx" name = "viperx"
def __init__( def __init__(
self, self,
config: ViperXRobotConfig, config: ViperXConfig,
): ):
super().__init__(config) super().__init__(config)
self.config = config self.config = config

View File

@ -1,4 +1,2 @@
from .configuration_widowx import WidowXTeleopConfig from .config_widowx import WidowXConfig
from .teleop_widowx import WidowXTeleop from .widowx import WidowX
__all__ = ["WidowXTeleopConfig", "WidowXTeleop"]

View File

@ -21,5 +21,5 @@ from ..config import TeleoperatorConfig
@TeleoperatorConfig.register_subclass("widowx") @TeleoperatorConfig.register_subclass("widowx")
@dataclass @dataclass
class WidowXTeleopConfig(TeleoperatorConfig): class WidowXConfig(TeleoperatorConfig):
port: str # Port to connect to the arm port: str # Port to connect to the arm

View File

@ -26,20 +26,20 @@ from lerobot.common.motors.dynamixel import (
) )
from ..teleoperator import Teleoperator from ..teleoperator import Teleoperator
from .configuration_widowx import WidowXTeleopConfig from .config_widowx import WidowXConfig
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class WidowXTeleop(Teleoperator): class WidowX(Teleoperator):
""" """
[WidowX](https://www.trossenrobotics.com/widowx-250) developed by Trossen Robotics [WidowX](https://www.trossenrobotics.com/widowx-250) developed by Trossen Robotics
""" """
config_class = WidowXTeleopConfig config_class = WidowXConfig
name = "widowx" name = "widowx"
def __init__(self, config: WidowXTeleopConfig): def __init__(self, config: WidowXConfig):
super().__init__(config) super().__init__(config)
self.config = config self.config = config
self.arm = DynamixelMotorsBus( self.arm = DynamixelMotorsBus(