diff --git a/.actions/assistant.py b/.actions/assistant.py index 63f8bb80e..73e8de5a7 100644 --- a/.actions/assistant.py +++ b/.actions/assistant.py @@ -320,7 +320,7 @@ def bash_render(folder: str) -> str: else: cmd.append(f"# available: {AssistantCLI.DEVICE_ACCELERATOR}\n") if AssistantCLI._valid_accelerator(folder): - cmd.append(f"python -m papermill.cli {ipynb_file} {pub_ipynb} --kernel python") + cmd.append(f"python -m papermill {ipynb_file} {pub_ipynb} --kernel python") else: warn("Invalid notebook's accelerator for this device. So no outputs will be generated.", RuntimeWarning) cmd.append(f"cp {ipynb_file} {pub_ipynb}") diff --git a/.azure/ipynb-publish.yml b/.azure/ipynb-publish.yml index 1349cd285..9e1ceea06 100644 --- a/.azure/ipynb-publish.yml +++ b/.azure/ipynb-publish.yml @@ -64,7 +64,7 @@ jobs: - bash: | set -e python -c "import torch ; mgpu = torch.cuda.device_count() ; assert mgpu > 0, f'GPU: {mgpu}'" - python -m papermill.cli --version + python -m papermill --version displayName: 'Sanity check' - bash: | diff --git a/requirements/devel.txt b/requirements/devel.txt index 1d0bfd73e..98fdc52a1 100644 --- a/requirements/devel.txt +++ b/requirements/devel.txt @@ -2,7 +2,7 @@ virtualenv jupytext # converting pytest>=6.0 nbval # testing -papermill # render +papermill>=2.3.4 # render black flake8 isort