Cleaning configs

lerobot/common/envs/configs.py

@@ -229,55 +229,6 @@ class HILSerlRobotEnvConfig(EnvConfig):
         return {}
 
 
-@EnvConfig.register_subclass("maniskill_push")
-@dataclass
-class ManiskillEnvConfig(EnvConfig):
-    """Configuration for the ManiSkill environment."""
-
-    name: str = "maniskill/pushcube"
-    task: str = "PushCube-v1"
-    image_size: int = 64
-    control_mode: str = "pd_ee_delta_pose"
-    state_dim: int = 25
-    action_dim: int = 7
-    fps: int = 200
-    episode_length: int = 50
-    obs_type: str = "rgb"
-    render_mode: str = "rgb_array"
-    render_size: int = 64
-    device: str = "cuda"
-    robot: str = "so100"  # This is a hack to make the robot config work
-    video_record: VideoRecordConfig = field(default_factory=VideoRecordConfig)
-    wrapper: WrapperConfig = field(default_factory=WrapperConfig)
-    mock_gripper: bool = False
-    features: dict[str, PolicyFeature] = field(
-        default_factory=lambda: {
-            "action": PolicyFeature(type=FeatureType.ACTION, shape=(7,)),
-            "observation.image": PolicyFeature(type=FeatureType.VISUAL, shape=(3, 64, 64)),
-            "observation.state": PolicyFeature(type=FeatureType.STATE, shape=(25,)),
-        }
-    )
-    features_map: dict[str, str] = field(
-        default_factory=lambda: {
-            "action": ACTION,
-            "observation.image": OBS_IMAGE,
-            "observation.state": OBS_ROBOT,
-        }
-    )
-    reward_classifier_pretrained_path: Optional[str] = None
-
-    @property
-    def gym_kwargs(self) -> dict:
-        return {
-            "obs_type": self.obs_type,
-            "render_mode": self.render_mode,
-            "max_episode_steps": self.episode_length,
-            "control_mode": self.control_mode,
-            "sensor_configs": {"width": self.image_size, "height": self.image_size},
-            "num_envs": 1,
-        }
-
-
 @EnvConfig.register_subclass("hil")
 @dataclass
 class HILEnvConfig(EnvConfig):

pyproject.toml

@@ -63,7 +63,6 @@ dependencies = [
     "omegaconf>=2.3.0",
     "opencv-python-headless>=4.9.0",
     "packaging>=24.2",
-    "protobuf>=5.29.3",
     "av>=14.2.0",
     "pymunk>=6.6.0",
     "pynput>=1.7.7",
@@ -72,9 +71,7 @@ dependencies = [
     "termcolor>=2.4.0",
     "torch>=2.2.1,<2.7",
     "torchcodec==0.2.1; sys_platform != 'win32' and (sys_platform != 'linux' or (platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')) and (sys_platform != 'darwin' or platform_machine != 'x86_64')",
-    "torchmetrics>=1.6.0",
     "torchvision>=0.21.0",
-    "transformers>=4.47.0",
     "wandb>=0.16.3",
     "zarr>=2.17.0",
 ]
@@ -88,7 +85,7 @@ dora = [
 ]
 dynamixel = ["dynamixel-sdk>=3.7.31", "pynput>=1.7.7"]
 feetech = ["feetech-servo-sdk>=1.0.0", "pynput>=1.7.7"]
-hilserl = ["transformers>=4.48.0", "torchmetrics>=1.6.0", "gym-hil>=0.1.2"]
+hilserl = ["transformers>=4.48.0", "torchmetrics>=1.6.0", "gym-hil>=0.1.2", "protobuf>=5.29.3"]
 intelrealsense = ["pyrealsense2>=2.55.1.6486 ; sys_platform != 'darwin'"]
 pi0 = ["transformers>=4.48.0"]
 pusht = ["gym-pusht>=0.1.5 ; python_version < '4.0'"]