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

bugfix: policy_kwargs can be None

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-26 13:03:03 +01:00
parent 9db4e93fa0
commit 5fa351db22
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

@ -114,6 +114,8 @@ class TRPL(BetterOnPolicyAlgorithm):
_init_setup_model: bool = True,
):
self.projection_class = castProjection(projection_class)
if projection_kwargs is None:
projection_kwargs = {}
self.projection_kwargs = projection_kwargs
self.projection = self.projection_class(**self.projection_kwargs)
if policy_kwargs is None: