From 5a2f85048ddc392fdcc0ea5d785fbccd779aadcf Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Tue, 9 Jan 2018 18:09:05 -0800 Subject: [PATCH] [rllib] Fix incorrect documentation on how to use custom models #1405 --- doc/source/rllib.rst | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) 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