improve crfmnes

google · Nov 13, 2022 · b46cc0d · b46cc0d
1 parent f1506b8
commit b46cc0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evojax/algo/crfmnes.py b/evojax/algo/crfmnes.py
@@ -61,7 +61,7 @@ def ask(self) -> jnp.ndarray:
         return self.jnp_stack(self.params)
 
     def tell(self, fitness: jnp.ndarray) -> None:
-        self.crfm.tell(-jnp.array(fitness))
+        self.crfm.tell(-fitness)
         self._best_params = self.crfm.x_best
 
     @property