From 57889d71a8e76544f45623b5df3bb122ee29f811 Mon Sep 17 00:00:00 2001 From: lc1021 Date: Thu, 19 Sep 2024 15:46:12 +0000 Subject: [PATCH] Move cmaes.py from core to baselines --- qdax/{core => baselines}/cmaes.py | 0 qdax/core/emitters/cma_emitter.py | 2 +- qdax/core/emitters/cma_mega_emitter.py | 2 +- qdax/core/emitters/cma_rnd_emitter.py | 2 +- tests/core_test/cmaes_test.py | 2 +- 5 files changed, 4 insertions(+), 4 deletions(-) rename qdax/{core => baselines}/cmaes.py (100%) diff --git a/qdax/core/cmaes.py b/qdax/baselines/cmaes.py similarity index 100% rename from qdax/core/cmaes.py rename to qdax/baselines/cmaes.py diff --git a/qdax/core/emitters/cma_emitter.py b/qdax/core/emitters/cma_emitter.py index 315dcd9b..e3b476dd 100644 --- a/qdax/core/emitters/cma_emitter.py +++ b/qdax/core/emitters/cma_emitter.py @@ -7,7 +7,7 @@ import jax import jax.numpy as jnp -from qdax.core.cmaes import CMAES, CMAESState +from qdax.baselines.cmaes import CMAES, CMAESState from qdax.core.containers.mapelites_repertoire import ( MapElitesRepertoire, get_cells_indices, diff --git a/qdax/core/emitters/cma_mega_emitter.py b/qdax/core/emitters/cma_mega_emitter.py index c3f87fed..976f528b 100644 --- a/qdax/core/emitters/cma_mega_emitter.py +++ b/qdax/core/emitters/cma_mega_emitter.py @@ -6,7 +6,7 @@ import jax import jax.numpy as jnp -from qdax.core.cmaes import CMAES, CMAESState +from qdax.baselines.cmaes import CMAES, CMAESState from qdax.core.containers.mapelites_repertoire import ( MapElitesRepertoire, get_cells_indices, diff --git a/qdax/core/emitters/cma_rnd_emitter.py b/qdax/core/emitters/cma_rnd_emitter.py index 27e4f0db..0715c437 100644 --- a/qdax/core/emitters/cma_rnd_emitter.py +++ b/qdax/core/emitters/cma_rnd_emitter.py @@ -6,7 +6,7 @@ import jax import jax.numpy as jnp -from qdax.core.cmaes import CMAESState +from qdax.baselines.cmaes import CMAESState from qdax.core.containers.mapelites_repertoire import MapElitesRepertoire from qdax.core.emitters.cma_emitter import CMAEmitter, CMAEmitterState from qdax.custom_types import Descriptor, ExtraScores, Fitness, Genotype, RNGKey diff --git a/tests/core_test/cmaes_test.py b/tests/core_test/cmaes_test.py index 16321fd4..daa7ce9d 100644 --- a/tests/core_test/cmaes_test.py +++ b/tests/core_test/cmaes_test.py @@ -4,7 +4,7 @@ import jax.numpy as jnp import pytest -from qdax.core.cmaes import CMAES +from qdax.baselines.cmaes import CMAES def test_cmaes() -> None: