From c9ca5c0b53b522a5a375dbc9a5cd769d1ba4c05b Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Mon, 21 Aug 2023 23:49:19 +0200 Subject: [PATCH] Fix: Exception when trying to use PCA --- sbBrix/common/policies.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sbBrix/common/policies.py b/sbBrix/common/policies.py index df1c1cc..66de5c8 100644 --- a/sbBrix/common/policies.py +++ b/sbBrix/common/policies.py @@ -580,6 +580,10 @@ class ActorCriticPolicy(BasePolicy): self.action_net, self.log_std = self.action_dist.proba_distribution_net( latent_dim=latent_dim_pi, latent_sde_dim=latent_dim_pi, log_std_init=self.log_std_init ) + elif isinstance(self.action_dist, PCA_Distribution): + self.action_net, self.log_std = self.action_dist.proba_distribution_net( + latent_dim=latent_dim_pi + ) elif isinstance(self.action_dist, (CategoricalDistribution, MultiCategoricalDistribution, BernoulliDistribution)): self.action_net = self.action_dist.proba_distribution_net(latent_dim=latent_dim_pi) else: