diff --git a/evojax/algo/crfmnes.py b/evojax/algo/crfmnes.py index 2ff1698a..6fae3fd6 100644 --- 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