import qdax.custom_types instead of qdax.types in notebooks

adaptive-intelligent-robotics · Sep 5, 2024 · f0b2d64 · f0b2d64
1 parent d7fd89e
commit f0b2d64
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/examples/aurora.ipynb b/examples/aurora.ipynb
@@ -93,7 +93,7 @@
     "from qdax.core.emitters.mutation_operators import isoline_variation\n",
     "from qdax.core.emitters.standard_emitters import MixingEmitter\n",
     "\n",
-    "from qdax.types import Observation\n",
+    "from qdax.custom_types import Observation\n",
     "from qdax.utils import train_seq2seq\n",
     "\n",
     "\n",

diff --git a/examples/jumanji_snake.ipynb b/examples/jumanji_snake.ipynb
@@ -72,7 +72,7 @@
     "from qdax.core.emitters.mutation_operators import isoline_variation\n",
     "\n",
     "from qdax.core.emitters.standard_emitters import MixingEmitter\n",
-    "from qdax.types import ExtraScores, Fitness, RNGKey, Descriptor\n",
+    "from qdax.custom_types import ExtraScores, Fitness, RNGKey, Descriptor\n",
     "from qdax.utils.metrics import default_ga_metrics, default_qd_metrics"
    ]
   },

diff --git a/examples/me_sac_pbt.ipynb b/examples/me_sac_pbt.ipynb
@@ -54,7 +54,7 @@
     "from qdax.core.distributed_map_elites import DistributedMAPElites\n",
     "from qdax.core.emitters.pbt_me_emitter import PBTEmitter, PBTEmitterConfig\n",
     "from qdax.core.emitters.pbt_variation_operators import sac_pbt_variation_fn\n",
-    "from qdax.types import Descriptor, ExtraScores, Fitness, Genotype, RNGKey\n",
+    "from qdax.custom_types import Descriptor, ExtraScores, Fitness, Genotype, RNGKey\n",
     "from qdax.utils.metrics import CSVLogger, default_qd_metrics\n",
     "from qdax.utils.plotting import plot_map_elites_results"
    ]

diff --git a/examples/me_td3_pbt.ipynb b/examples/me_td3_pbt.ipynb
@@ -56,7 +56,7 @@
     "from qdax.core.emitters.pbt_me_emitter import PBTEmitter, PBTEmitterConfig\n",
     "from qdax.core.emitters.pbt_variation_operators import td3_pbt_variation_fn\n",
     "from qdax.core.distributed_map_elites import DistributedMAPElites\n",
-    "from qdax.types import RNGKey\n",
+    "from qdax.custom_types import RNGKey\n",
     "from qdax.utils.metrics import default_qd_metrics\n",
     "from qdax.utils.plotting import plot_2d_map_elites_repertoire, plot_map_elites_results"
    ]

diff --git a/examples/mome.ipynb b/examples/mome.ipynb
@@ -81,7 +81,7 @@
     "\n",
     "import matplotlib.pyplot as plt\n",
     "\n",
-    "from qdax.types import Fitness, Descriptor, RNGKey, ExtraScores"
+    "from qdax.custom_types import Fitness, Descriptor, RNGKey, ExtraScores"
    ]
   },
   {

diff --git a/examples/nsga2_spea2.ipynb b/examples/nsga2_spea2.ipynb
@@ -88,7 +88,7 @@
     "from qdax.utils.plotting import plot_global_pareto_front\n",
     "from qdax.utils.metrics import default_ga_metrics\n",
     "\n",
-    "from qdax.types import Genotype, Fitness, Descriptor"
+    "from qdax.custom_types import Genotype, Fitness, Descriptor"
    ]
   },
   {

diff --git a/examples/pga_aurora.ipynb b/examples/pga_aurora.ipynb
@@ -93,7 +93,7 @@
     "from qdax.core.emitters.mutation_operators import isoline_variation\n",
     "from qdax.core.emitters.pga_me_emitter import PGAMEConfig, PGAMEEmitter\n",
     "\n",
-    "from qdax.types import Observation\n",
+    "from qdax.custom_types import Observation\n",
     "from qdax.utils import train_seq2seq\n",
     "\n",
     "\n",
-Original file line number
+Diff line change
@@ Expand Up / @@ -81,7 +81,7 @@ @@
         "\n",
         "import matplotlib.pyplot as plt\n",
         "\n",
-        "from qdax.types import Fitness, Descriptor, RNGKey, ExtraScores"
+        "from qdax.custom_types import Fitness, Descriptor, RNGKey, ExtraScores"
        ]
       },
       {
@@ Expand Down @@