是否可以使用造纸厂停止执行 Jupyter 笔记本？

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 中。