Changeset View
Changeset View
Standalone View
Standalone View
python_modules/libraries/dagstermill/dagstermill/engine.py
import nbformat | import nbformat | ||||
from papermill.log import logger | from papermill.log import logger | ||||
from .compat import IS_PAPERMILL_2, ExecutionError | from .compat import ExecutionError, is_papermill_2 | ||||
if IS_PAPERMILL_2: | if is_papermill_2(): | ||||
# pylint: disable=import-error,no-name-in-module | # pylint: disable=import-error,no-name-in-module | ||||
from papermill.clientwrap import PapermillNotebookClient | from papermill.clientwrap import PapermillNotebookClient | ||||
from papermill.engines import NBClientEngine # pylint: disable=import-error | from papermill.engines import NBClientEngine # pylint: disable=import-error | ||||
from papermill.utils import merge_kwargs, remove_args | from papermill.utils import merge_kwargs, remove_args | ||||
class DagstermillNotebookClient(PapermillNotebookClient): | class DagstermillNotebookClient(PapermillNotebookClient): | ||||
def papermill_execute_cells(self): | def papermill_execute_cells(self): | ||||
try: | try: | ||||
▲ Show 20 Lines • Show All 124 Lines • Show Last 20 Lines |