diff --git a/doc/source/rllib.rst b/doc/source/rllib.rst index c10b5cc33..337e643e3 100644 --- a/doc/source/rllib.rst +++ b/doc/source/rllib.rst @@ -183,9 +183,11 @@ custom model and preprocessor classes are documented in the ray.init() alg = ppo.PPOAgent(env="CartPole-v0", config={ - "custom_preprocessor": "my_prep", - "custom_model": "my_model", - "custom_options": {}, # extra options to pass to your classes + "model": { + "custom_preprocessor": "my_prep", + "custom_model": "my_model", + "custom_options": {}, # extra options to pass to your classes + }, }) Using RLlib with Ray.tune