Make sure pca_policy knowns, we are using pca

2023-08-22 01:10:41 +02:00 · 2023-08-22 01:10:41 +02:00 · 0af83eb7dc
commit 0af83eb7dc
parent d6c05668cb
1 changed files with 1 additions and 0 deletions
--- 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,