From d9d6f1a3a12770bf6866ca78bef7ff3f1637167e Mon Sep 17 00:00:00 2001
From: Henry <henry.webel@cpr.ku.dk>
Date: Thu, 7 Sep 2023 17:19:59 +0200
Subject: [PATCH] :memo: add hint to restart kernel in colab (tutorial)

-> next is to speed up install time by removing upper dependency bounds
---
 project/04_1_train_pimms_models.ipynb | 10 +++++++++-
 project/04_1_train_pimms_models.py    |  5 ++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/project/04_1_train_pimms_models.ipynb b/project/04_1_train_pimms_models.ipynb
index 93a1aeb76..21ac7cd0f 100644
--- a/project/04_1_train_pimms_models.ipynb
+++ b/project/04_1_train_pimms_models.ipynb
@@ -32,9 +32,17 @@
     "    !pip install pimms-learn"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "36b5a27d",
+   "metadata": {},
+   "source": [
+    "If on colab, please restart the environment and run everything from here on."
+   ]
+  },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "id": "65b5bdaf",
    "metadata": {},
    "outputs": [],
diff --git a/project/04_1_train_pimms_models.py b/project/04_1_train_pimms_models.py
index 603e0b783..fea106857 100644
--- a/project/04_1_train_pimms_models.py
+++ b/project/04_1_train_pimms_models.py
@@ -16,7 +16,10 @@
 if IN_COLAB:
     print("Install PIMMS...")
     # # !pip install git+https://github.com/RasmussenLab/pimms.git@dev
-    # # !pip install pimms-learn
+    # !pip install pimms-learn
+
+# %% [markdown]
+# If on colab, please restart the environment and run everything from here on.
 
 # %%
 import os