From 148118cbe3150811da0c3090583df0e746021907 Mon Sep 17 00:00:00 2001
From: N_T <nils.thuerey @ tum.de>
Date: Mon, 9 Dec 2024 14:24:48 +0800
Subject: [PATCH] improved pip install for probmodels-ddpm-fm

---
 probmodels-ddpm-fm.ipynb | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/probmodels-ddpm-fm.ipynb b/probmodels-ddpm-fm.ipynb
index a4dd102..b96a25a 100644
--- a/probmodels-ddpm-fm.ipynb
+++ b/probmodels-ddpm-fm.ipynb
@@ -76,7 +76,12 @@
     }
    ],
    "source": [
-    "%pip install --upgrade --quiet einops bayesian_torch\n",
+    "try:\n",
+    "    import google.colab  # to ensure that we are inside colab\n",
+    "    !pip install --upgrade --quiet einops bayesian_torch\n",
+    "except ImportError:\n",
+    "    print('This notebook is running locally, please make sure the necessary pip packages are installed.')\n",
+    "    pass\n",
     "!git clone https://github.com/tum-pbs/Diffusion-based-Flow-Prediction.git\n",
     "%cd Diffusion-based-Flow-Prediction/"
    ]