bug fix in ppo constructor

lasgroup · Nov 10, 2023 · f4ec063 · f4ec063
1 parent 351569a
commit f4ec063
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mbpo/optimizers/policy_optimizers/brax_optimizers.py b/mbpo/optimizers/policy_optimizers/brax_optimizers.py
@@ -102,6 +102,7 @@ def train(self,
 class PPOOptimizer(BraxOptimizer):
     def __init__(self,
                  true_buffer: UniformSamplingQueue,
+                 system: System | None = None,
                  **ppo_kwargs):
         super().__init__(agent_class=PPO, system=system, true_buffer=true_buffer, **ppo_kwargs)