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

Fix: Crash when no dist_kwargs defined

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-03-09 13:51:38 +01:00
parent 8f66a34c29
commit 66b8d21398
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
metastable_baselines2/common/on_policy_algorithm.py
View File

@ -92,6 +92,8 @@ 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) policy_kwargs['dist_kwargs']['n_envs'] = len(env.envs)
super().__init__( super().__init__(