python - 是否可以使用 papermill 停止执行 Jupyter notebook？

Question

我有一个设置，其中我有一些参数化的 Jupyter 笔记本，因此我可以从另一个笔记本运行它们。

我正在使用 papermill 模块来执行此操作，这非常方便。最简单的运行方式看起来像这样

path = '/path/to/notebook.ipynb'
pm.execute_notebook(
   path,
   path,    
)

我现在想要的是能够在不满足某些条件时从正在运行的笔记本中停止执行。

有没有办法用 papermill 做到这一点，如果没有，是否有另一个模块可以做到这一点？

Answer 1

我只会在笔记本中使用 assert not failure_condition, "My condition failed" 来强制引发异常。这将停止执行并引发异常，papermill 将在调用 pm.execute_notebook 的父进程中包装在 PapermillExecutionError 中。