adding colab setup to notebooks

graphcore-research · Apr 30, 2024 · bc04124 · bc04124
1 parent 4c98d6d
commit bc04124
Show file tree

Hide file tree

Showing 3 changed files with 303 additions and 25 deletions.
diff --git a/docs/batching.ipynb b/docs/batching.ipynb
@@ -35,7 +35,28 @@
     "the batch we build a `Hamiltonian` for each bond length and stack the built modules to \n",
     "create a batched-Hamiltonian.  This example uses the `sto-3g` basis set and the simple\n",
     "local density approximation of density functional theory but the this formulation isn't\n",
-    "unique to these choices for how the Hamiltonian is represented."
+    "unique to these choices for how the Hamiltonian is represented.\n",
+    "\n",
+    ":::{note}\n",
+    "The following code cell will install MESS into the Google Colab runtime.\n",
+    "Select the 🚀 in the toolbar above to try this out!\n",
+    ":::"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "tags": [
+     "hide-cell"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "import sys\n",
+    "\n",
+    "if 'google.colab' in sys.modules:\n",
+    "    !pip install git+https://github.com/graphcore-research/mess.git\n"
    ]
   },
   {

diff --git a/docs/optim.ipynb b/docs/optim.ipynb
diff --git a/docs/tour.ipynb b/docs/tour.ipynb
@@ -30,7 +30,7 @@
         "object will display a 3D visualisation\n",
         "\n",
         ":::{note}\n",
-        "The following code cell will install mess into the Google Colab runtime.\n",
+        "The following code cell will install MESS into the Google Colab runtime.\n",
         "Select the 🚀 in the toolbar above to try this out!\n",
         ":::"
       ]