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,