From 531dee3a8219011de8e7ae3fdb5433d8d65e64b7 Mon Sep 17 00:00:00 2001 From: Aleksandr Mokrov Date: Mon, 27 Jan 2025 21:46:44 +0100 Subject: [PATCH] Fix broken tensorflow uninstallation (#2694) CVS-161267 --- .../whisper-subtitles-generation.ipynb | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/notebooks/whisper-subtitles-generation/whisper-subtitles-generation.ipynb b/notebooks/whisper-subtitles-generation/whisper-subtitles-generation.ipynb index 36f74d416fa..2d5f2845f54 100644 --- 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" ] },