Fixed issue in RND emitter with es_params not passed to ES

adaptive-intelligent-robotics · Nov 24, 2023 · 2b99584 · 2b99584
1 parent a97a95f
commit 2b99584
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qdax/core/emitters/cma_me_policies.py b/qdax/core/emitters/cma_me_policies.py
@@ -343,7 +343,7 @@ def init(
         self.es = Strategies[self.es_type](
             num_dims=self.reshaper.genotype_dim,
             popsize=self.batch_size,
-            sigma_init=self._sigma_g,
+            **self.es_params,
         )
 
         # Initialize repertoire with default values