Merge pull request #439 from openvinotoolkit/update_post_inference_ho…

…ok_notebook Add check to make sure project is trained in notebook 012
openvinotoolkit · Jun 12, 2024 · 30009ae · 30009ae
2 parents d19cc9f + a5b7eaa
commit 30009ae
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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)"
    ]
   },
   {