From a5b7eaa34535ff61d8fe8b1ba05e356d3c29db3c Mon Sep 17 00:00:00 2001
From: ljcornel <ludo.cornelissen@intel.com>
Date: Wed, 12 Jun 2024 15:51:49 +0200
Subject: [PATCH] Add check to make sure project is trained in notebook 012

---
 notebooks/012_post_inference_hooks.ipynb | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/notebooks/012_post_inference_hooks.ipynb b/notebooks/012_post_inference_hooks.ipynb
index 72f11079..30df9df9 100644
--- a/notebooks/012_post_inference_hooks.ipynb
+++ b/notebooks/012_post_inference_hooks.ipynb
@@ -49,8 +49,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from geti_sdk.demos import ensure_trained_example_project\n",
+    "\n",
     "PROJECT_NAME = \"COCO animal detection demo\"\n",
-    "project = geti.get_project(PROJECT_NAME)"
+    "project = ensure_trained_example_project(geti=geti, project_name=PROJECT_NAME)"
    ]
   },
   {