fix notebook conversion files by timeout (openvinotoolkit#1625)

nikita-savelyevv · Jan 18, 2024 · 0e19acb · 0e19acb
1 parent 69eb976
commit 0e19acb
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/.ci/convert_notebooks.py b/.ci/convert_notebooks.py
@@ -58,8 +58,12 @@ def main():
         start = time.perf_counter()
         print(f"Convert {notebook_path}")
         if str(notebook_path) not in ignore_execution_list:
-            retcode = subprocess.run(["jupyter", "nbconvert",  "--log-level=INFO", "--execute", "--to",  "notebook", "--output",
-                                     str(notebook_executed),  '--output-dir', str(root), '--ExecutePreprocessor.kernel_name=python3', str(notebook_path)], timeout=args.timeout).returncode
+            try:
+                retcode = subprocess.run(["jupyter", "nbconvert",  "--log-level=INFO", "--execute", "--to",  "notebook", "--output",
+                                        str(notebook_executed),  '--output-dir', str(root), '--ExecutePreprocessor.kernel_name=python3', str(notebook_path)], timeout=args.timeout).returncode
+            except subprocess.TimeoutExpired:
+                retcode = -42
+                print(f"TIMEOUT: {notebook_path}")
             if retcode:
                 failed_notebooks.append(str(notebook_path))
                 continue

diff --git a/.ci/ignore_convert_execution.txt b/.ci/ignore_convert_execution.txt
@@ -26,23 +26,26 @@ notebooks/236-stable-diffusion-v2/236-stable-diffusion-v2-optimum-demo-compariso
 notebooks/236-stable-diffusion-v2/236-stable-diffusion-v2-optimum-demo.ipynb
 notebooks/236-stable-diffusion-v2/236-stable-diffusion-v2-text-to-image-demo.ipynb
 notebooks/237-segment-anything/237-segment-anything.ipynb
-notebooks/237-segment-anything/237-segment-anything-fiftyone.ipynb
-notebooks/238-deepfloyd-if/238-deep-floyd-if.ipynb
+notebooks/238-deepfloyd-if/238-deep-floyd-if-convert.ipynb
 notebooks/238-deepfloyd-if/238-deep-floyd-if-optimize.ipynb
 notebooks/239-image-bind/239-image-bind-convert.ipynb
-notebooks/239-image-bind/239-image-bind-convert.ipynb
+notebooks/239-image-bind/239-image-bind-quantize.ipynb
 notebooks/240-dolly-2-instruction-following/240-dolly-2-instruction-following.ipynb
 notebooks/241-riffusion-text-to-music/241-riffusion-text-to-music.ipynb
 notebooks/244-named-entity-recognition/244-named-entity-recognition.ipynb
 notebooks/248-stable-diffusion-xl/248-stable-diffusion-xl.ipynb
+notebooks/248-stable-diffusion-xl/248-segmind-vegart.ipynb
+notebooks/248-stable-diffusion-xl/248-ssd-b1.ipynb
 notebooks/249-oneformer-segmentation/249-oneformer-segmentation.ipynb
 notebooks/251-tiny-sd-image-generation/251-tiny-sd-image-generation.ipynb
 notebooks/252-fastcomposer-image-generation/252-fastcomposer-image-generation.ipynb
 notebooks/253-zeroscope-text2video/253-zeroscope-text2video.ipynb
 notebooks/254-llm-chatbot/254-llm-chatbot.ipynb
+notebooks/254-llm-chatbot/254-rag-chatbot.ipynb
 notebooks/255-mms-massively-multilingual-speech/255-mms-massively-multilingual-speech.ipynb
 notebooks/256-bark-text-to-audio/256-bark-text-to-audio.ipynb
 notebooks/257-llava-multimodal-chatbot/257-llava-multimodal-chatbot.ipynb
+notebooks/257-llava-multimodal-chatbot/257-videollava-multimodal-chatbot.ipynb
 notebooks/258-blip-diffusion-subject-generation/258-blip-diffusion-subject-generation.ipynb
 notebooks/259-decidiffusion-image-generation/259-decidiffusion-image-generation.ipynb
 notebooks/260-pix2struct-docvqa/260-pix2struct-docvqa.ipynb
@@ -58,4 +61,5 @@ notebooks/270-sound-generation-audioldm2/270-sound-generation-audioldm2.ipynb
 notebooks/271-sdxl-turbo/271-sdxl-turbo.ipynb
 notebooks/272-paint-by-example/272-paint-by-example.ipynb
 notebooks/273-stable-zephyr-3b-chatbot/273-stable-zephyr-3b-chatbot.ipynb
+notebooks/275-llm-question-answering/275-llm-question-answering.ipynb
 notebooks/404-style-transfer-webcam/404-style-transfer.ipynb