From 0af83eb7dc4d5e5d69a834031608790a57dbbb93 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Tue, 22 Aug 2023 01:10:41 +0200
Subject: [PATCH] Make sure pca_policy knowns, we are using pca

---
 sbBrix/common/off_policy_algorithm.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sbBrix/common/off_policy_algorithm.py b/sbBrix/common/off_policy_algorithm.py
index 4d5e0dd..816d033 100644
--- a/sbBrix/common/off_policy_algorithm.py
+++ b/sbBrix/common/off_policy_algorithm.py
@@ -85,6 +85,7 @@ class BetterOffPolicyAlgorithm(OffPolicyAlgorithm):
     ):
         assert not (use_sde and use_pca)
         self.use_pca = use_pca
+        policy_kwargs["use_pca"] = self.use_pca
         super().__init__(
             policy=policy,
             env=env,