Skip to content

Commit

Permalink
Limit testing notebooks
Browse files Browse the repository at this point in the history
  • Loading branch information
yatarkan committed May 30, 2024
1 parent 60b6248 commit 3cccbfa
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions .ci/validate_notebooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,6 +122,16 @@ def prepare_test_plan(test_list: Optional[List[str]], ignore_list: List[str], nb
"Testing notebooks should be provided to '--test_list' argument as a txt file or should be empty to test all notebooks.\n"
f"Received test list: {test_list}"
)
# TODO Remove after testing
notebooks_to_check = [
"amused-lightweight-text-to-image.ipynb",
"detectron2-to-openvino.ipynb",
"meter-reader.ipynb",
"openvino-tokenizers.ipynb",
"s3d-mil-nce-text-to-video-retrieval.ipynb",
"table-question-answering.ipynb",
]
testing_notebooks = list(filter(lambda tn: any(n in str(tn) for n in notebooks_to_check), testing_notebooks))
testing_notebooks = sorted(list(set(testing_notebooks)))
print(f"Testing notebooks: {testing_notebooks}")

Expand Down

0 comments on commit 3cccbfa

Please sign in to comment.