From 8384d322877eb733ea0295a2e9307a522d0f0506 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Tue, 22 Aug 2023 00:10:54 +0200
Subject: [PATCH] SAC support for PCA

---
 sbBrix/sac/sac.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sbBrix/sac/sac.py b/sbBrix/sac/sac.py
index b3ee34e..52213c2 100644
--- a/sbBrix/sac/sac.py
+++ b/sbBrix/sac/sac.py
@@ -106,6 +106,7 @@ class SAC(BetterOffPolicyAlgorithm):
         use_sde: bool = False,
         sde_sample_freq: int = -1,
         use_sde_at_warmup: bool = False,
+        use_pca: bool = False,
         stats_window_size: int = 100,
         tensorboard_log: Optional[str] = None,
         policy_kwargs: Optional[Dict[str, Any]] = None,
@@ -137,6 +138,7 @@ class SAC(BetterOffPolicyAlgorithm):
             use_sde=use_sde,
             sde_sample_freq=sde_sample_freq,
             use_sde_at_warmup=use_sde_at_warmup,
+            use_pca=use_pca,
             optimize_memory_usage=optimize_memory_usage,
             supported_action_spaces=(spaces.Box),
             support_multi_env=True,