Skip to content
Snippets Groups Projects
Commit 16d1e63e authored by Fionn O'Donohoe's avatar Fionn O'Donohoe
Browse files

[notebooks] increase notebook timeout to 1 hour


Signed-off-by: default avatarFionn O'Donohoe <fionno@xilinx.com>
parent 30501464
No related branches found
No related tags found
No related merge requests found
......@@ -5,6 +5,7 @@ from nbconvert.preprocessors import ExecutePreprocessor
from finn.util.basic import get_finn_root
notebook_timeout_seconds = 3600
notebook_basic_dir = get_finn_root() + "/notebooks/basics/"
notebook_advanced_dir = get_finn_root() + "/notebooks/advanced/"
notebook_cyber_dir = get_finn_root() + "/notebooks/end2end_example/cybersecurity/"
......@@ -76,7 +77,9 @@ bnn_notebooks = [
def test_notebook_exec(notebook):
with open(notebook) as f:
nb = nbformat.read(f, as_version=4)
ep = ExecutePreprocessor(timeout=600, kernel_name="python3")
ep = ExecutePreprocessor(
timeout=notebook_timeout_seconds, kernel_name="python3"
)
try:
assert ep.preprocess(nb) is not None, f"Got empty notebook for {notebook}"
except Exception:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment