diff --git a/rllib/examples/replay_buffer_api.py b/rllib/examples/replay_buffer_api.py index ea9e30b46ebd..4ef20ea7b82f 100644 --- a/rllib/examples/replay_buffer_api.py +++ b/rllib/examples/replay_buffer_api.py @@ -53,7 +53,7 @@ .environment(env="CartPole-v0") .training(model=dict(use_lstm=True, lstm_cell_size=64, max_seq_len=20)) .framework(framework=args.framework) - .rollouts(num_workers=4) + .rollouts(num_rollout_workers=4) ) stop_config = {