[RLlib] fix an example (ray-project#34325)

Signed-off-by: Rohan Potdar <[email protected]>
Rohan138 · Apr 25, 2023 · ce021b3 · ce021b3
1 parent 9a5c547
commit ce021b3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rllib/examples/cartpole_lstm.py b/rllib/examples/cartpole_lstm.py
@@ -105,7 +105,8 @@
  # >>
  # >> while True:
  # >> a, state_out, _ = algo.compute_single_action(
- # .. obs, state, prev_a, prev_r)
+ # .. obs, state, prev_action=prev_a, prev_reward=prev_r
+ # .. )
  # >> obs, reward, done, truncated, _ = env.step(a)
  # >> if done:
  # >> obs, info = env.reset()