[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

examples/notebooks/Brax_Experiments_Visualization.ipynb

@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "fc7b28eb-db2b-4a02-87d4-dfc44a9fecc2",
+   "id": "0",
    "metadata": {},
    "source": [
     "# Visualization of the `brax` Experiment Results\n",
@@ -13,7 +13,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d5350a77-c84b-4d05-9e1c-4a6e0e2f6d0d",
+   "id": "1",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -24,7 +24,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e4c4d0df-0480-4c5f-a4ab-d34727286cc1",
+   "id": "2",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -34,7 +34,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d64f3d04-680a-43c5-a807-dbaf9f917198",
+   "id": "3",
    "metadata": {},
    "source": [
     "---"
@@ -43,7 +43,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7025f8b3-9982-4196-9790-9e8c1491dc06",
+   "id": "4",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -55,7 +55,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d5bccb74-a64a-4e55-9ae9-9686f772fa60",
+   "id": "5",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -66,7 +66,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0d71c7c3-bb36-47fb-88dc-bd285c54c44a",
+   "id": "6",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -77,7 +77,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "54c7bee4-bf5c-445c-b009-40484391461c",
+   "id": "7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -89,7 +89,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "30eca014-be71-4162-b675-0eaf8b037bce",
+   "id": "8",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -100,7 +100,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "78ce1f2f-cb89-432e-ad98-07f7e1d5d457",
+   "id": "9",
    "metadata": {},
    "source": [
     "Below, we put the values of the center solution into the policy network as parameters:"
@@ -109,7 +109,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "545d693d-ccd9-4ea7-b068-06b832f32be5",
+   "id": "10",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -118,7 +118,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e7c7581d-24ba-4f06-88e2-41bb3274cd37",
+   "id": "11",
    "metadata": {},
    "source": [
     "---\n",
@@ -131,7 +131,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "8d798a71-003f-4bed-9f12-391dc7b797c0",
+   "id": "12",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -143,7 +143,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "30080d4b-227c-4bde-8009-7a97c6e177f1",
+   "id": "13",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -153,7 +153,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c64a6482-0533-40c8-b8f9-1b1ad3ee0019",
+   "id": "14",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -163,7 +163,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ee69ba99-bcb7-4d21-9f5c-6c330cb3ed96",
+   "id": "15",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -183,7 +183,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c99f0ea4-7638-4e8a-a0a2-d9f483c8c096",
+   "id": "16",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -194,7 +194,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "19d38fe1-9d6f-40ec-926d-3c4066a4b66a",
+   "id": "17",
    "metadata": {},
    "source": [
     "Below, we define a utility function named `use_policy(...)`.\n",
@@ -214,7 +214,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e8c6554d-4bdb-49b9-8c2e-956bce6ddb8a",
+   "id": "18",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -238,7 +238,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "28c044fe-4639-411c-be78-ac09cbe5e78f",
+   "id": "19",
    "metadata": {},
    "source": [
     "We now initialize a new instance of our brax environment, and trigger the jit compilation on its `reset` and `step` methods."
@@ -247,7 +247,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "c1251e82-1d0f-4e43-a6c5-1ec4c0275dab",
+   "id": "20",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -258,7 +258,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "55229cc2-aad5-4c78-b095-010a538adb40",
+   "id": "21",
    "metadata": {},
    "source": [
     "Below we run our policy and collect the states of the episodes."
@@ -267,7 +267,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "220ec837-1d1a-401e-a144-5516bdb3e493",
+   "id": "22",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -291,7 +291,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c9bbe53a-40f5-4bf5-a063-47f1d88032d6",
+   "id": "23",
    "metadata": {},
    "source": [
     "Length of the episode and the total reward:"
@@ -300,7 +300,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "80345e9c-0694-413d-81c2-205dfacdfb5a",
+   "id": "24",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -310,7 +310,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "8d2e2506-5bb2-4b1f-bb00-004465008fa3",
+   "id": "25",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -325,7 +325,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "8b6ec424-e6cc-453a-93fd-eb07e44c1bd6",
+   "id": "26",
    "metadata": {},
    "source": [
     "Visualization of the policy:"
@@ -334,7 +334,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "29b9c3aa-8068-40bc-9ec2-e5ff759ed22c",
+   "id": "27",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -349,7 +349,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b50e36a7-dd33-4fc3-9f9d-2d7dca93110b",
+   "id": "28",
    "metadata": {},
    "outputs": [],
    "source": [