Skip to content

Commit fb3981b

Browse files
correct the error msg for invalid env registering
Signed-off-by: Ashwin <ashwinsnambiar12@gmail.com>
1 parent 6ecf992 commit fb3981b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

rllib/utils/error.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@ class NotSerializable(Exception):
4949
a) For Atari support: `pip install gym[atari] autorom[accept-rom-license]`.
5050
For PyBullet support: `pip install pybullet`.
5151
b) To register your custom env, do `from ray import tune;
52-
tune.register('[name]', lambda cfg: [return env obj from here using cfg])`.
53-
Then in your config, do `config['env'] = [name]`.
52+
tune.register_env('[name]', lambda cfg: [return env obj from here using cfg])`.
53+
Then in your config, do `config.environment(env='[name]').
5454
c) Make sure you provide a fully qualified classpath, e.g.:
5555
`ray.rllib.examples.envs.classes.repeat_after_me_env.RepeatAfterMeEnv`
5656
"""

0 commit comments

Comments
 (0)