Merge pull request #1 from lasgroup/dev/kiten

Added iCEM optimizer
lasgroup · Nov 6, 2024 · 8429bdd · 8429bdd
2 parents 99acfb8 + 92283fb
commit 8429bdd
Show file tree

Hide file tree

Showing 3 changed files with 355 additions and 195 deletions.
diff --git a/mbpo/optimizers/__init__.py b/mbpo/optimizers/__init__.py
@@ -1,7 +1,6 @@
 from mbpo.optimizers.policy_optimizers.sac.sac import SAC
-from mbpo.optimizers.trajectory_optimizers.icem_optimizer import iCemTO, iCemOptimizerState, iCemParams
+from mbpo.optimizers.trajectory_optimizers.icem_optimizer import iCemTO, iCemOptimizerState, iCemParams, iCEMOptimizer
 from mbpo.optimizers.base_optimizer import BaseOptimizer
 from mbpo.optimizers.policy_optimizers.brax_optimizers import PPOOptimizer, SACOptimizer, BraxOptimizer, \
     BraxState, BraxOutput
 from mbpo.optimizers.policy_optimizers.bptt_optimizer import BPTTOptimizer, BPTTState
-