fix last imports orders

adaptive-intelligent-robotics · Aug 26, 2024 · 86c7f9a · 86c7f9a
1 parent 0e61d6b
commit 86c7f9a
Show file tree

Hide file tree

Showing 29 changed files with 126 additions and 84 deletions.
diff --git a/examples/jumanji_snake.ipynb b/examples/jumanji_snake.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "233e0f03",
+   "id": "0",
    "metadata": {},
    "source": [
     "# Training a population on Jumanji-Snake with QDax\n",
@@ -15,7 +15,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "47b46c2f",
+   "id": "1",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -78,7 +78,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "03c2f1f7",
+   "id": "2",
    "metadata": {},
    "source": [
     "## Define hyperparameters"
@@ -87,7 +87,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "52dd1e3b",
+   "id": "3",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -105,7 +105,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8b8c890a",
+   "id": "4",
    "metadata": {},
    "source": [
     "## Instantiate the snake environment"
@@ -114,7 +114,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a842cccc",
+   "id": "5",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -132,7 +132,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "776862f1",
+   "id": "6",
    "metadata": {},
    "source": [
     "## Define the type of policy that will be used to solve the problem"
@@ -141,7 +141,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2a1ce7d0",
+   "id": "7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -161,7 +161,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "49586b07",
+   "id": "8",
    "metadata": {},
    "source": [
     "## Utils to interact with the environment\n",
@@ -172,7 +172,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d1ff7827",
+   "id": "9",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -219,7 +219,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0078bc01",
+   "id": "10",
    "metadata": {},
    "source": [
     "## Init a population of policies\n",
@@ -230,7 +230,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6cbd2065",
+   "id": "11",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -255,7 +255,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "fe6bf07f",
+   "id": "12",
    "metadata": {},
    "source": [
     "## Define a method to extract behavior descriptor when relevant"
@@ -264,7 +264,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a264b672",
+   "id": "13",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -311,7 +311,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1cdc5f87",
+   "id": "14",
    "metadata": {},
    "source": [
     "## Define the scoring function"
@@ -320,7 +320,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7b77d826",
+   "id": "15",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -333,7 +333,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6555491a",
+   "id": "16",
    "metadata": {},
    "source": [
     "## Define the emitter used"
@@ -342,7 +342,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "30061ff4",
+   "id": "17",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -360,7 +360,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "da7e9b74",
+   "id": "18",
    "metadata": {},
    "source": [
     "## Define the algorithm used and apply the initial step\n",
@@ -371,7 +371,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f7b5c2d6",
+   "id": "19",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -415,7 +415,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "9b1bfee5",
+   "id": "20",
    "metadata": {},
    "source": [
     "## Run the optimization loop"
@@ -424,7 +424,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d1af3a35",
+   "id": "21",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -442,7 +442,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "114ea4a8",
+   "id": "22",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -452,7 +452,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "92a35bf0",
+   "id": "23",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -462,7 +462,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "79ada2d5",
+   "id": "24",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -472,7 +472,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "fe5da301",
+   "id": "25",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -489,7 +489,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "93d8154e",
+   "id": "26",
    "metadata": {},
    "source": [
     "## Play snake with the best policy\n",
@@ -500,7 +500,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3ff882f4",
+   "id": "27",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -511,7 +511,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "762c167e",
+   "id": "28",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -524,7 +524,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "07523e33",
+   "id": "29",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -537,7 +537,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c75ce088",
+   "id": "30",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -550,7 +550,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "50ef95f6",
+   "id": "31",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -563,7 +563,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "40a03409",
+   "id": "32",
    "metadata": {},
    "outputs": [],
    "source": [