diff --git a/docs/source/env_processor.mdx b/docs/source/env_processor.mdx index cfe7df8d4..c3563b5cc 100644 --- a/docs/source/env_processor.mdx +++ b/docs/source/env_processor.mdx @@ -88,7 +88,7 @@ policy_preprocessor = NormalizerProcessorStep(stats=dataset_stats) The same policy can work with different environment processors, and the same environment processor can work with different policies: -```python +````python # Use SmolVLA policy with LIBERO environment libero_preprocessor, libero_postprocessor = make_env_pre_post_processors(libero_cfg) smolvla_preprocessor, smolvla_postprocessor = make_pre_post_processors(smolvla_cfg) @@ -139,7 +139,7 @@ class LiberoVelocityProcessorStep(ObservationProcessorStep): state = torch.cat([eef_pos, eef_axisangle, eef_vel, gripper_pos, gripper_vel], dim=-1) # 14D return state -``` +```` ### 4. **Cleaner Environment Code** @@ -325,7 +325,7 @@ class MyEnvProcessorStep(ObservationProcessorStep): ### 2. Update the Factory -```python +````python # In src/lerobot/envs/factory.py def make_env_pre_post_processors(env_cfg: EnvConfig): @@ -364,7 +364,7 @@ lerobot-eval \ --policy.path=lerobot/my_policy \ --env.type=myenv \ # Automatically uses MyEnvProcessorStep --eval.n_episodes=10 -``` +```` ## Future: Environment Postprocessors