From d2501f12444220fefae5c3928abad57996114578 Mon Sep 17 00:00:00 2001 From: Joseph Suarez Date: Fri, 22 Nov 2024 20:58:44 +0000 Subject: [PATCH] remove framework from cleanrl --- docs/docs.html | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/docs.html b/docs/docs.html index 2e2cab0..5f0a14e 100644 --- a/docs/docs.html +++ b/docs/docs.html @@ -339,12 +339,12 @@

Policies

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)