From 5fa351db2270f49c7f8cf13e89ed87a7e575fda8 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Fri, 26 Jan 2024 13:03:03 +0100
Subject: [PATCH] bugfix: policy_kwargs can be None

---
 metastable_baselines2/trpl/trpl.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/metastable_baselines2/trpl/trpl.py b/metastable_baselines2/trpl/trpl.py
index 83d78ab..94a020f 100644
--- a/metastable_baselines2/trpl/trpl.py
+++ b/metastable_baselines2/trpl/trpl.py
@@ -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: