diff --git a/docs/source/adding_benchmarks.mdx b/docs/source/adding_benchmarks.mdx index db599bb3c..2788eca6e 100644 --- a/docs/source/adding_benchmarks.mdx +++ b/docs/source/adding_benchmarks.mdx @@ -138,7 +138,7 @@ Each `EnvConfig` subclass declares two dicts that tell the policy what to expect Create a `gym.Env` subclass that wraps the third-party simulator: ```python -class MyBenchmarkEnv(gym.Env): +class MyBenchmarkConfig(EnvConfig): metadata = {"render_modes": ["rgb_array"], "render_fps": } def __init__(self, task_suite, task_id, ...):