From ce021b3651df27b5cae0d21b7d308929a691cb5c Mon Sep 17 00:00:00 2001 From: Rohan Potdar <66227218+Rohan138@users.noreply.github.com> Date: Tue, 25 Apr 2023 16:36:46 -0400 Subject: [PATCH] [RLlib] fix an example (#34325) Signed-off-by: Rohan Potdar --- rllib/examples/cartpole_lstm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rllib/examples/cartpole_lstm.py b/rllib/examples/cartpole_lstm.py index 1f5cb0128fd45..ce01295380f98 100644 --- 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()