From 4a7fbc2470f3d0adc89d41538070ad7e9d1a8dc8 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 9 Mar 2024 13:57:52 +0100 Subject: [PATCH] Fix typo --- metastable_baselines2/common/policies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metastable_baselines2/common/policies.py b/metastable_baselines2/common/policies.py index 43e6a54..18e9428 100644 --- a/metastable_baselines2/common/policies.py +++ b/metastable_baselines2/common/policies.py @@ -530,7 +530,7 @@ class ActorCriticPolicy(BasePolicy): self.n_envs = dist_kwargs.pop('n_envs', 1) # Action distribution - self.action_dist = make_proba_distribution(action_space, self.n_envs, use_sde=use_sde, use_pca=use_pca, n_envs=self.n_envs, dist_kwargs=dist_kwargs) + self.action_dist = make_proba_distribution(action_space,use_sde=use_sde, use_pca=use_pca, n_envs=self.n_envs, dist_kwargs=dist_kwargs) self._build(lr_schedule)