From f1f8eb10d2a63bcd9301e3ff690e40e498905fd4 Mon Sep 17 00:00:00 2001 From: Alex Parsons Date: Mon, 7 Aug 2023 16:16:56 +0000 Subject: [PATCH] Make updating run notebook optional --- src/data_common/management/run_notebook.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/data_common/management/run_notebook.py b/src/data_common/management/run_notebook.py index 27058aa..0cb12eb 100644 --- a/src/data_common/management/run_notebook.py +++ b/src/data_common/management/run_notebook.py @@ -3,7 +3,7 @@ from pathlib import Path -def run_notebook(notebook_filename: Path): +def run_notebook(notebook_filename: Path, save: bool = True): """ Run a notebook as part of another process """ @@ -12,6 +12,8 @@ def run_notebook(notebook_filename: Path): nb = nbformat.read(f, as_version=4) ep = ExecutePreprocessor(timeout=600) ep.preprocess(nb, {"metadata": {"path": "notebooks/"}}) - with open(notebook_filename, "w", encoding="utf-8") as f: - nbformat.write(nb, f) + if save: + print(f"Saving notebook: {notebook_filename}") + with open(notebook_filename, "w", encoding="utf-8") as f: + nbformat.write(nb, f) print("Done")