From c00fc5169fc1b7fef8c9e0ed8dc37e144c6b857d Mon Sep 17 00:00:00 2001
From: Vitor Shen <17490173+shenvitor@users.noreply.github.com>
Date: Wed, 14 Aug 2024 11:16:29 +0200
Subject: [PATCH] test1

---
 docs/lambda-k-pi/automated.ipynb | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/docs/lambda-k-pi/automated.ipynb b/docs/lambda-k-pi/automated.ipynb
index 20f95dc..b1595e5 100644
--- a/docs/lambda-k-pi/automated.ipynb
+++ b/docs/lambda-k-pi/automated.ipynb
@@ -19,9 +19,6 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
-    "jupyter": {
-     "source_hidden": true
-    },
     "mystnb": {
      "code_prompt_show": "Import Python libraries"
     },
@@ -31,10 +28,27 @@
    },
    "outputs": [],
    "source": [
+    "# ruff: noqa: S404, S603, S607\n",
+    "\n",
     "from __future__ import annotations\n",
     "\n",
     "import logging\n",
     "import os\n",
+    "\n",
+    "if \"COLAB_BACKEND_VERSION\" in os.environ:\n",
+    "    import subprocess\n",
+    "\n",
+    "    from google.colab import output\n",
+    "\n",
+    "    dependencies = [\n",
+    "        \"ampform\",\n",
+    "        \"graphviz\",\n",
+    "        \"ipywidgets\",\n",
+    "        \"ipympl\",\n",
+    "        \"tensorwaves[jax,pwa]\",\n",
+    "    ]\n",
+    "    subprocess.check_call([\"pip\", \"install\", *dependencies])\n",
+    "    output.enable_custom_widget_manager()\n",
     "import warnings\n",
     "from collections import defaultdict\n",
     "from fractions import Fraction\n",
@@ -766,7 +780,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.5"
+   "version": "3.12.4"
   }
  },
  "nbformat": 4,