From 43418a5e536979268d07cd284d45e3fb92bea964 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 21 May 2023 16:40:13 +0200 Subject: [PATCH] Tiny Bugfix --- metastable_baselines/misc/rollout_buffer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metastable_baselines/misc/rollout_buffer.py b/metastable_baselines/misc/rollout_buffer.py index 58d4d42..2b68ffe 100644 --- a/metastable_baselines/misc/rollout_buffer.py +++ b/metastable_baselines/misc/rollout_buffer.py @@ -213,7 +213,7 @@ class GaussianRolloutCollectorAuxclass(): with th.no_grad(): # Convert to pytorch tensor or to TensorDict obs_tensor = obs_as_tensor(self._last_obs, self.device) - if self.policy['use_pca']: + if self.policy.use_pca: actions, values, log_probs = self.policy(obs_tensor, trajectory=self.get_past_trajectories()) else: actions, values, log_probs = self.policy(obs_tensor)