From 6d776f7eb99d0cc6a5310f5862865253fa4a5dee Mon Sep 17 00:00:00 2001 From: fracapuano Date: Sat, 31 May 2025 21:07:20 +0200 Subject: [PATCH] minor fixes iterating on steven's comments --- lerobot/__init__.py | 7 +------ pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/lerobot/__init__.py b/lerobot/__init__.py index f8acafce3..632698128 100644 --- a/lerobot/__init__.py +++ b/lerobot/__init__.py @@ -168,12 +168,7 @@ available_datasets = sorted( ) # lists all available policies from `lerobot/common/policies` -available_policies = [ - "act", - "diffusion", - "tdmpc", - "vqbet", -] +available_policies = ["act", "diffusion", "tdmpc", "vqbet", "smolvla"] # lists all available robots from `lerobot/common/robot_devices/robots` available_robots = [ diff --git a/pyproject.toml b/pyproject.toml index 0265c5f7f..b32b9ddee 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -86,7 +86,7 @@ dynamixel = ["dynamixel-sdk>=3.7.31", "pynput>=1.7.7"] feetech = ["feetech-servo-sdk>=1.0.0", "pynput>=1.7.7"] intelrealsense = ["pyrealsense2>=2.55.1.6486 ; sys_platform != 'darwin'"] pi0 = ["transformers>=4.48.0"] -smolvla = ["transformers>=4.50.3", "pytest>=8.3.5", "num2words>=0.5.14", "accelerate>=1.7.0"] +smolvla = ["transformers>=4.50.3", "num2words>=0.5.14", "accelerate>=1.7.0"] pusht = ["gym-pusht>=0.1.5 ; python_version < '4.0'"] stretch = [ "hello-robot-stretch-body>=0.7.27 ; python_version < '4.0' and sys_platform == 'linux'",