Add file to run all notebooks

Author: ColCarroll

scripts/run_notebooks.py

@@ -8,20 +8,25 @@
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 NOTEBOOK_DIR = os.path.join(BASE_DIR, 'docs', 'source', 'notebooks')
 
+
 def list_notebooks():
     """Get an iterator of filepaths to notebooks in NOTEBOOK_DIR"""
     return glob.glob(os.path.join(NOTEBOOK_DIR, '*.ipynb'))
 
+
 def execute_notebook(notebook_path):
     """Run and overwrite a notebook file."""
     ep = ExecutePreprocessor(timeout=-1)
     with open(notebook_path, 'r') as buff:
         nb = nbformat.read(buff, as_version=nbformat.NO_CONVERT)
     try:
         t0 = time.time()
-        _ = ep.preprocess(nb, {'metadata': {'path': NOTEBOOK_DIR}})
+        ep.preprocess(nb, {'metadata': {'path': NOTEBOOK_DIR}})
         t1 = time.time()
 
+    except KeyboardInterrupt:
+        raise
+
     except BaseException as e:
         t1 = time.time()
         return False, 'Failed after {:.1f}s:\n{}'.format(t1 - t0, str(e))
@@ -31,8 +36,9 @@ def execute_notebook(notebook_path):
 
     return True, 'Succeeded after {:.1f}s'.format(t1 - t0)
 
+
 def run_all_notebooks():
-    """Tries to re-run all notebooks.  Prints failures at end."""
+    """Try to re-run all notebooks.  Print failures at end."""
     failed = []
     for notebook_path in list_notebooks():
         print('Executing {}'.format(os.path.basename(notebook_path)))
@@ -46,6 +52,5 @@ def run_all_notebooks():
         print('\n'.join(failed))
 
 
-
 if __name__ == '__main__':
     run_all_notebooks()