Tiny fix for other envs

2022-11-07 13:23:55 +01:00
parent 302dbf6dde
commit 4ec5c65cf2
2 changed files with 35 additions and 33 deletions
@@ -136,6 +136,8 @@ class ActorCriticPolicy(BasePolicy):
        self.normalize_images = normalize_images
        self.log_std_init = log_std_init
        # Keyword arguments for gSDE distribution
+        if dist_kwargs == None:
+            dist_kwargs = {}
        if use_sde:
            add_dist_kwargs = {
                'use_sde': True,