chore: Move cmaes.py from core to baselines (#200)

* Move cmaes.py from core to baselines * Change corresponding api docs --------- Authored-by: Lisa <[email protected]>
adaptive-intelligent-robotics · Sep 20, 2024 · b2ef13e · b2ef13e
1 parent 14f30f3
commit b2ef13e
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 7 deletions.
diff --git a/docs/api_documentation/core/cmaes.md b/docs/api_documentation/core/cmaes.md
@@ -1,3 +1,3 @@
 # CMAES class
 
-::: qdax.core.cmaes.CMAES
+::: qdax.baselines.cmaes.CMAES
diff --git a/examples/cmaes.ipynb b/examples/cmaes.ipynb
@@ -55,7 +55,7 @@
     "import matplotlib.pyplot as plt\n",
     "from matplotlib.patches import Ellipse\n",
     "\n",
-    "from qdax.core.cmaes import CMAES"
+    "from qdax.baselines.cmaes import CMAES"
    ]
   },
   {

diff --git a/mkdocs.yml b/mkdocs.yml
@@ -154,11 +154,11 @@ nav:
         - DADS: api_documentation/core/dads.md
         - SAC: api_documentation/core/sac.md
         - TD3: api_documentation/core/td3.md
-        - CMAES: api_documentation/core/cmaes.md
         - Genetic Algorithm: api_documentation/core/genetic_algorithm.md
         - NSGA2: api_documentation/core/nsga2.md
         - SPEA2: api_documentation/core/spea2.md
         - PBT: api_documentation/core/pbt.md
+        - CMAES: api_documentation/core/cmaes.md
       - Containers: api_documentation/core/containers.md
       - Emitters: api_documentation/core/emitters.md
       - Neuroevolution: api_documentation/core/neuroevolution.md

diff --git a/qdax/core/cmaes.py → qdax/baselines/cmaes.py b/qdax/core/cmaes.py → qdax/baselines/cmaes.py
diff --git 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
@@ -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
@@ -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
@@ -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: