Fix broken tensorflow uninstallation (#2694)

CVS-161267
openvinotoolkit · Jan 27, 2025 · 531dee3 · 531dee3
1 parent f165f1a
commit 531dee3
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/notebooks/whisper-subtitles-generation/whisper-subtitles-generation.ipynb b/notebooks/whisper-subtitles-generation/whisper-subtitles-generation.ipynb
@@ -98,11 +98,14 @@
     "\n",
     "from packaging import version\n",
     "\n",
-    "if (\n",
-    "    importlib.util.find_spec(\"tensorflow\") is not None\n",
-    "    and version.parse(importlib.metadata.version(\"tensorflow\")) < version.parse(\"2.18.0\")\n",
-    "    and version.parse(importlib.metadata.version(\"numpy\")) >= version.parse(\"2.0.0\")\n",
-    "):\n",
+    "try:\n",
+    "    if (\n",
+    "        importlib.util.find_spec(\"tensorflow\") is not None\n",
+    "        and version.parse(importlib.metadata.version(\"tensorflow\")) < version.parse(\"2.18.0\")\n",
+    "        and version.parse(importlib.metadata.version(\"numpy\")) >= version.parse(\"2.0.0\")\n",
+    "    ):\n",
+    "        %pip uninstall -q -y tensorflow\n",
+    "except importlib.metadata.PackageNotFoundError:\n",
     "    %pip uninstall -q -y tensorflow"
    ]
   },