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")
@dataclass
class ViperXRobotConfig(RobotConfig):
class ViperXConfig(RobotConfig):
port: str # Port to connect to the arm
disable_torque_on_disconnect: bool = True

View File

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

View File

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

View File

@ -21,5 +21,5 @@ from ..config import TeleoperatorConfig
@TeleoperatorConfig.register_subclass("widowx")
@dataclass
class WidowXTeleopConfig(TeleoperatorConfig):
class WidowXConfig(TeleoperatorConfig):
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 .configuration_widowx import WidowXTeleopConfig
from .config_widowx import WidowXConfig
logger = logging.getLogger(__name__)
class WidowXTeleop(Teleoperator):
class WidowX(Teleoperator):
"""
[WidowX](https://www.trossenrobotics.com/widowx-250) developed by Trossen Robotics
"""
config_class = WidowXTeleopConfig
config_class = WidowXConfig
name = "widowx"
def __init__(self, config: WidowXTeleopConfig):
def __init__(self, config: WidowXConfig):
super().__init__(config)
self.config = config
self.arm = DynamixelMotorsBus(