dodox/metastable-baselines2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix problem when env is provided just as id

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-03-14 17:34:47 +01:00
parent 1c3d3cf6cf
commit 39c21ab6b9
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
metastable_baselines2/common/on_policy_algorithm.py
View File

@ -92,10 +92,6 @@ class BetterOnPolicyAlgorithm(OnPolicyAlgorithm):
assert not rollout_buffer_class and not rollout_buffer_kwargs assert not rollout_buffer_class and not rollout_buffer_kwargs
if 'dist_kwargs' not in policy_kwargs:
policy_kwargs['dist_kwargs'] = {}
policy_kwargs['dist_kwargs']['n_envs'] = len(env.envs)
super().__init__( super().__init__(
policy=policy, policy=policy,
env=env, env=env,
@ -122,6 +118,10 @@ class BetterOnPolicyAlgorithm(OnPolicyAlgorithm):
_init_setup_model=_init_setup_model _init_setup_model=_init_setup_model
) )
if 'dist_kwargs' not in self.policy_kwargs:
self.policy_kwargs['dist_kwargs'] = {}
self.policy_kwargs['dist_kwargs']['n_envs'] = len(self.env.envs)
self.rollout_buffer_class = None self.rollout_buffer_class = None
self.rollout_buffer_kwargs = {} self.rollout_buffer_kwargs = {}