Skip to content

Commit

Permalink
remove framework from cleanrl
Browse files Browse the repository at this point in the history
  • Loading branch information
Joseph Suarez committed Nov 22, 2024
1 parent f77bd7b commit d2501f1
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docs/docs.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -339,12 +339,12 @@ <h1>Policies</h1>
actions, value, state = lstm_policy(obs, state)

# Use our CleanRL API compatibility layer
import pufferlib.frameworks.cleanrl
cleanrl_policy = pufferlib.frameworks.cleanrl.Policy(policy)
import pufferlib.cleanrl
cleanrl_policy = pufferlib.cleanrl.Policy(policy)
actions = cleanrl_policy.get_action_and_value(obs)[0].numpy()

# Use our CleanRL LSTM API compatibility layer
cleanrl_lstm_policy = pufferlib.frameworks.cleanrl.RecurrentPolicy(lstm_policy)
cleanrl_lstm_policy = pufferlib.cleanrl.RecurrentPolicy(lstm_policy)
actions = cleanrl_lstm_policy.get_action_and_value(obs)[0].numpy()

obs, rewards, terminals, truncateds, infos = vecenv.step(actions)
Expand Down

0 comments on commit d2501f1

Please sign in to comment.