A script to time running Jupyter notebooks

runner.py

import nbformat
from nbconvert.preprocessors import ExecutePreprocessor

import sys

import time

notebook_filename = sys.argv[1]

t0 = time.time()

with open(notebook_filename) as f:
    nb = nbformat.read(f, as_version=4)

t1 = time.time()

ep = ExecutePreprocessor(timeout=1, kernel_name='python3')

t2 = time.time()

ep.preprocess(nb, {'metadata': {'path': 'notebooks/'}})

t3 = time.time()

with open('executed_notebook.ipynb', 'wt') as f:
    nbformat.write(nb, f)

t4 = time.time()

print(t1-t0, t2-t1, t3-t2, t4-t3)