Make sure pca_policy knowns, we are using pca

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