Lint code

Author: maxencefaldor

examples/cmaes.ipynb

@@ -25,15 +25,15 @@
   },
   {
    "cell_type": "markdown",
-   "id": "931313f8",
+   "id": "2",
    "metadata": {},
    "source": [
     "## Installation"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "d43f4db2",
+   "id": "3",
    "metadata": {},
    "source": [
     "You will need Python 3.11 or later, and a working JAX installation. For example, you can install JAX with:"
@@ -42,7 +42,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ad841f6f",
+   "id": "4",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -51,7 +51,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "366885ad",
+   "id": "5",
    "metadata": {},
    "source": [
     "Then, install QDax from PyPI:"
@@ -60,7 +60,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "07a2fbdc",
+   "id": "6",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -70,7 +70,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "3",
+   "id": "7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -85,7 +85,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4",
+   "id": "8",
    "metadata": {},
    "source": [
     "## Set the hyperparameters"
@@ -94,7 +94,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "5",
+   "id": "9",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -112,7 +112,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6",
+   "id": "10",
    "metadata": {
     "pycharm": {
      "name": "#%% md\n"
@@ -125,7 +125,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7",
+   "id": "11",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -147,7 +147,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8",
+   "id": "12",
    "metadata": {
     "pycharm": {
      "name": "#%% md\n"
@@ -160,7 +160,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9",
+   "id": "13",
    "metadata": {
     "pycharm": {
      "name": "#%%\n"
@@ -181,7 +181,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "10",
+   "id": "14",
    "metadata": {
     "pycharm": {
      "name": "#%% md\n"
@@ -194,7 +194,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "11",
+   "id": "15",
    "metadata": {
     "pycharm": {
      "name": "#%%\n"
@@ -208,7 +208,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "12",
+   "id": "16",
    "metadata": {
     "pycharm": {
      "name": "#%% md\n"
@@ -221,7 +221,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "13",
+   "id": "17",
    "metadata": {
     "pycharm": {
      "name": "#%%\n"
@@ -261,7 +261,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "14",
+   "id": "18",
    "metadata": {},
    "source": [
     "## Check final fitnesses and distribution mean"
@@ -270,7 +270,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "15",
+   "id": "19",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -288,7 +288,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "16",
+   "id": "20",
    "metadata": {
     "pycharm": {
      "name": "#%% md\n"
@@ -301,7 +301,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "17",
+   "id": "21",
    "metadata": {
     "pycharm": {
      "name": "#%%\n"