Можно ли остановить работу ноутбука Jupyter на бумажной фабрике?

Question

У меня есть настройка, при которой у меня есть несколько ноутбуков Jupyter, которые параметризованы, чтобы я мог запускать их из другого ноутбука.

Я использую модуль papermill, чтобы сделать это, что довольно удобно,Самый простой способ запуска выглядит примерно так:

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

Теперь мне бы хотелось иметь возможность останавливать выполнение из записной книжки , если определенные условия не выполняются.

Есть ли способ сделать это с помощью бумажной фабрики, а если нет, то есть ли другой модуль, который может это сделать?

Answer 1

Я бы просто использовал assert not failure_condition, "My condition failed" внутри ноутбука, чтобы вызвать исключение.Это остановит выполнение и создаст исключение, что бумажная фабрика обернет PapermillExecutionError в родительском процессе, который вызвал pm.execute_notebook.