Add an exception if cloning TripoSR repository is failed (#2397)

CVS-152581
openvinotoolkit · Sep 17, 2024 · 3889abd · 3889abd
1 parent d37453e
commit 3889abd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/notebooks/triposr-3d-reconstruction/triposr-3d-reconstruction.ipynb b/notebooks/triposr-3d-reconstruction/triposr-3d-reconstruction.ipynb
@@ -100,11 +100,15 @@
     }
    ],
    "source": [
+    "import os\n",
     "import sys\n",
     "from pathlib import Path\n",
     "\n",
     "if not Path(\"TripoSR\").exists():\n",
-    "    !git clone https://huggingface.co/spaces/stabilityai/TripoSR\n",
+    "    exit_code = os.system(\"git clone https://huggingface.co/spaces/stabilityai/TripoSR\")\n",
+    "\n",
+    "    if exit_code != 0:\n",
+    "        raise Exception(\"Failed to clone the repository!\")\n",
     "\n",
     "sys.path.append(\"TripoSR\")"
    ]