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

Fix issue when policy_kwargs=None and TRPL used

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-26 13:01:36 +01:00
parent e788e9f998
commit 9db4e93fa0
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/trpl/trpl.py
View File

@ -116,6 +116,8 @@ class TRPL(BetterOnPolicyAlgorithm):
self.projection_class = castProjection(projection_class)
self.projection_kwargs = projection_kwargs
self.projection = self.projection_class(**self.projection_kwargs)
if policy_kwargs is None:
policy_kwargs = {}
policy_kwargs['policy_projection'] = self.projection
super().__init__(