From f8cf7900769cce5b718a93d325f72316661806ab Mon Sep 17 00:00:00 2001 From: Roland Mas Date: Wed, 4 Dec 2024 15:11:38 +0100 Subject: [PATCH 01/10] Remove testsuite artefact --- papermill/tests/test_iorw.py | 1 + 1 file changed, 1 insertion(+) diff --git a/papermill/tests/test_iorw.py b/papermill/tests/test_iorw.py index ab09f01a..17961f58 100644 --- a/papermill/tests/test_iorw.py +++ b/papermill/tests/test_iorw.py @@ -206,6 +206,7 @@ def test_write_local_directory(self): with patch.object(io, 'open'): # Shouldn't raise with missing directory LocalHandler().write("buffer", "local.ipynb") + os.unlink("local.ipynb") def test_write_passed_cwd(self): with TemporaryDirectory() as temp_dir: From b8b635134023861d1789a1b97bd6adcdff983298 Mon Sep 17 00:00:00 2001 From: Aryan Karamtoth Date: Sat, 11 Apr 2026 09:56:26 +0100 Subject: [PATCH 02/10] Remove relative paths and entrypoints dependency Forwarded: no Last-Update: 2026-03-28 This patch is modified version of the 2 older patches that aren't compatible with the new release. This patch removes the relative paths used in the test files as well as the entrypoints dependency python-papermill (2.7.0-1) UNRELEASED; urgency=medium * Team Upload. * New Upstream Release 2.7.0 * Bump Standards-Version to 4.7.3 --- papermill/engines.py | 4 ++-- papermill/iorw.py | 4 ++-- papermill/tests/test_abs.py | 3 +-- papermill/tests/test_adl.py | 6 +++--- papermill/tests/test_autosave.py | 7 +++---- papermill/tests/test_cli.py | 4 ++-- papermill/tests/test_clientwrap.py | 6 +++--- papermill/tests/test_engines.py | 19 +++++++++++-------- papermill/tests/test_exceptions.py | 2 +- papermill/tests/test_execute.py | 12 ++++++------ papermill/tests/test_gcs.py | 4 ++-- papermill/tests/test_hdfs.py | 2 +- papermill/tests/test_iorw.py | 19 +++++++++++-------- papermill/tests/test_parameterize.py | 7 ++++--- papermill/tests/test_s3.py | 2 +- papermill/tests/test_translators.py | 6 +++--- papermill/tests/test_utils.py | 4 ++-- requirements.txt | 1 - 18 files changed, 58 insertions(+), 54 deletions(-) diff --git a/papermill/engines.py b/papermill/engines.py index c44d548d..c4659db2 100644 --- a/papermill/engines.py +++ b/papermill/engines.py @@ -5,7 +5,7 @@ from functools import wraps import dateutil -import entrypoints +from importlib.metadata import entry_points from .clientwrap import PapermillNotebookClient from .exceptions import PapermillException @@ -34,7 +34,7 @@ def register_entry_points(self): Load handlers provided by other packages """ - for entrypoint in entrypoints.get_group_all("papermill.engine"): + for entrypoint in entry_points().select(group="papermill.engine"): self.register(entrypoint.name, entrypoint.load()) def get_engine(self, name=None): diff --git a/papermill/iorw.py b/papermill/iorw.py index 14a0122c..8386e914 100644 --- a/papermill/iorw.py +++ b/papermill/iorw.py @@ -5,7 +5,7 @@ import warnings from contextlib import contextmanager -import entrypoints +from importlib.metadata import entry_points import nbformat import requests import yaml @@ -116,7 +116,7 @@ def register(self, scheme, handler): def register_entry_points(self): # Load handlers provided by other packages - for entrypoint in entrypoints.get_group_all("papermill.io"): + for entrypoint in entry_points().select(group="papermill.io"): self.register(entrypoint.name, entrypoint.load()) def get_handler(self, path, extensions=None): diff --git a/papermill/tests/test_abs.py b/papermill/tests/test_abs.py index 57dff73f..c383248d 100644 --- a/papermill/tests/test_abs.py +++ b/papermill/tests/test_abs.py @@ -4,8 +4,7 @@ from azure.identity import EnvironmentCredential -from ..abs import AzureBlobStore - +from papermill.abs import AzureBlobStore class MockBytesIO: def __init__(self): diff --git a/papermill/tests/test_adl.py b/papermill/tests/test_adl.py index 3aa98544..1c27f978 100644 --- a/papermill/tests/test_adl.py +++ b/papermill/tests/test_adl.py @@ -1,9 +1,9 @@ import unittest from unittest.mock import MagicMock, Mock, patch -from ..adl import ADL -from ..adl import core as adl_core -from ..adl import lib as adl_lib +from papermill.adl import ADL +from papermill.adl import core as adl_core +from papermill.adl import lib as adl_lib class ADLTest(unittest.TestCase): diff --git a/papermill/tests/test_autosave.py b/papermill/tests/test_autosave.py index 74ae06e8..6a0b0bf9 100644 --- a/papermill/tests/test_autosave.py +++ b/papermill/tests/test_autosave.py @@ -6,10 +6,9 @@ import nbformat -from .. import engines -from ..engines import NotebookExecutionManager -from ..execute import execute_notebook -from . import get_notebook_path +from papermill import engines +from papermill.engines import NotebookExecutionManager +from papermill.execute import execute_notebook class TestMidCellAutosave(unittest.TestCase): diff --git a/papermill/tests/test_cli.py b/papermill/tests/test_cli.py index 72ef5f9a..095ede2b 100755 --- a/papermill/tests/test_cli.py +++ b/papermill/tests/test_cli.py @@ -15,8 +15,8 @@ import pytest from click.testing import CliRunner -from .. import cli -from ..cli import _is_float, _is_int, _resolve_type, papermill +from papermill import cli +from papermill.cli import _is_float, _is_int, _resolve_type, papermill from . import get_notebook_path, kernel_name diff --git a/papermill/tests/test_clientwrap.py b/papermill/tests/test_clientwrap.py index cfa2a81a..6bf592f0 100644 --- a/papermill/tests/test_clientwrap.py +++ b/papermill/tests/test_clientwrap.py @@ -3,9 +3,9 @@ import nbformat -from ..clientwrap import PapermillNotebookClient -from ..engines import NotebookExecutionManager -from ..log import logger +from papermill.clientwrap import PapermillNotebookClient +from papermill.engines import NotebookExecutionManager +from papermill.log import logger from . import get_notebook_path diff --git a/papermill/tests/test_engines.py b/papermill/tests/test_engines.py index db5ee17c..a47f33de 100644 --- a/papermill/tests/test_engines.py +++ b/papermill/tests/test_engines.py @@ -6,10 +6,10 @@ import dateutil from nbformat.notebooknode import NotebookNode -from .. import engines, exceptions -from ..engines import Engine, NBClientEngine, NotebookExecutionManager -from ..iorw import load_notebook_node -from ..log import logger +from papermill import engines, exceptions +from papermill.engines import Engine, NBClientEngine, NotebookExecutionManager +from papermill.iorw import load_notebook_node +from papermill.log import logger from . import get_notebook_path @@ -489,10 +489,13 @@ def test_getting(self): self.assertRaises(exceptions.PapermillException, self.papermill_engines.get_engine, "non-existent") def test_registering_entry_points(self): - fake_entrypoint = Mock(load=Mock()) - fake_entrypoint.name = "fake-engine" + fake_entrypoint = Mock() + fake_entrypoint.load.return_value = Mock() - with patch("entrypoints.get_group_all", return_value=[fake_entrypoint]) as mock_get_group_all: + mock_entry_points = Mock() + mock_entry_points.select.return_value = [fake_entrypoint] + + with patch("papermill.engines.entry_points", return_value=mock_entry_points): self.papermill_engines.register_entry_points() - mock_get_group_all.assert_called_once_with("papermill.engine") + mock_entry_points.select.assert_called_once_with(group="papermill.engine") self.assertEqual(self.papermill_engines.get_engine("fake-engine"), fake_entrypoint.load.return_value) diff --git a/papermill/tests/test_exceptions.py b/papermill/tests/test_exceptions.py index 0a7e7a8d..6aaa72a9 100644 --- a/papermill/tests/test_exceptions.py +++ b/papermill/tests/test_exceptions.py @@ -4,7 +4,7 @@ import pytest -from .. import exceptions +from papermill import exceptions @pytest.fixture diff --git a/papermill/tests/test_execute.py b/papermill/tests/test_execute.py index 09600d64..c5abc95c 100644 --- a/papermill/tests/test_execute.py +++ b/papermill/tests/test_execute.py @@ -10,12 +10,12 @@ import nbformat from nbformat import validate -from .. import engines, translators -from ..exceptions import PapermillExecutionError, strip_color -from ..execute import execute_notebook -from ..iorw import load_notebook_node -from ..log import logger -from ..utils import chdir +from papermill import engines, translators +from papermill.exceptions import PapermillExecutionError +from papermill.execute import execute_notebook +from papermill.iorw import load_notebook_node +from papermill.log import logger +from papermill.utils import chdir from . import get_notebook_path, kernel_name execute_notebook = partial(execute_notebook, kernel_name=kernel_name) diff --git a/papermill/tests/test_gcs.py b/papermill/tests/test_gcs.py index ebd635a0..081f05d2 100644 --- a/papermill/tests/test_gcs.py +++ b/papermill/tests/test_gcs.py @@ -1,8 +1,8 @@ import unittest from unittest.mock import patch -from ..exceptions import PapermillRateLimitException -from ..iorw import GCSHandler, fallback_gs_is_retriable +from papermill.exceptions import PapermillRateLimitException +from papermill.iorw import GCSHandler, fallback_gs_is_retriable try: try: diff --git a/papermill/tests/test_hdfs.py b/papermill/tests/test_hdfs.py index 44c024df..1e5da20c 100644 --- a/papermill/tests/test_hdfs.py +++ b/papermill/tests/test_hdfs.py @@ -1,7 +1,7 @@ import unittest from unittest.mock import MagicMock, patch -from ..iorw import HDFSHandler +from papermill.iorw import HDFSHandler class MockHadoopFileSystem(MagicMock): diff --git a/papermill/tests/test_iorw.py b/papermill/tests/test_iorw.py index 17961f58..7608dc42 100644 --- a/papermill/tests/test_iorw.py +++ b/papermill/tests/test_iorw.py @@ -10,9 +10,9 @@ import pytest from requests.exceptions import ConnectionError -from .. import iorw -from ..exceptions import PapermillException -from ..iorw import ( +from papermill import iorw +from papermill.exceptions import PapermillException +from papermill.iorw import ( ADLHandler, HttpHandler, LocalHandler, @@ -101,12 +101,15 @@ def test_get_notebook_node_handler(self): self.assertIsInstance(self.papermill_io.get_handler(test_nb), NotebookNodeHandler) def test_entrypoint_register(self): - fake_entrypoint = Mock(load=Mock()) - fake_entrypoint.name = "fake-from-entry-point://" + fake_entrypoint = Mock() + fake_entrypoint.name = "fake-from-entry-point://" + fake_entrypoint.load.return_value = Mock() - with patch("entrypoints.get_group_all", return_value=[fake_entrypoint]) as mock_get_group_all: - self.papermill_io.register_entry_points() - mock_get_group_all.assert_called_once_with("papermill.io") + mock_entry_points = Mock() + mock_entry_points.select.return_value = [fake_entrypoint] + + with patch("papermill.iorw.entry_points", return_value=mock_entry_points): self.papermill_io.register_entry_points() + mock_entry_points.select.assert_called_once_with(group="papermill.io") fake_ = self.papermill_io.get_handler("fake-from-entry-point://") assert fake_ == fake_entrypoint.load.return_value diff --git a/papermill/tests/test_parameterize.py b/papermill/tests/test_parameterize.py index 431caa12..71e77433 100644 --- a/papermill/tests/test_parameterize.py +++ b/papermill/tests/test_parameterize.py @@ -1,9 +1,10 @@ import unittest from datetime import datetime -from ..exceptions import PapermillMissingParameterException -from ..iorw import load_notebook_node -from ..parameterize import add_builtin_parameters, parameterize_notebook, parameterize_path + +from papermill.exceptions import PapermillMissingParameterException +from papermill.iorw import load_notebook_node +from papermill.parameterize import add_builtin_parameters, parameterize_notebook, parameterize_path from . import get_notebook_path diff --git a/papermill/tests/test_s3.py b/papermill/tests/test_s3.py index bf006830..8c6391a1 100644 --- a/papermill/tests/test_s3.py +++ b/papermill/tests/test_s3.py @@ -7,7 +7,7 @@ import pytest from moto import mock_aws -from ..s3 import S3, Bucket, Key, Prefix +from papermill.s3 import S3, Bucket, Key, Prefix @pytest.fixture diff --git a/papermill/tests/test_translators.py b/papermill/tests/test_translators.py index 0edc1f07..95651bf1 100644 --- a/papermill/tests/test_translators.py +++ b/papermill/tests/test_translators.py @@ -4,9 +4,9 @@ import pytest from nbformat.v4 import new_code_cell -from .. import translators -from ..exceptions import PapermillException -from ..models import Parameter +from papermill import translators +from papermill.exceptions import PapermillException +from papermill.models import Parameter @pytest.mark.parametrize( diff --git a/papermill/tests/test_utils.py b/papermill/tests/test_utils.py index 4228b436..a6ef8a54 100644 --- a/papermill/tests/test_utils.py +++ b/papermill/tests/test_utils.py @@ -6,8 +6,8 @@ import pytest from nbformat.v4 import new_code_cell, new_notebook -from ..exceptions import PapermillParameterOverwriteWarning -from ..utils import ( +from papermill.exceptions import PapermillParameterOverwriteWarning +from papermill.utils import ( any_tagged_cell, chdir, merge_kwargs, diff --git a/requirements.txt b/requirements.txt index 6f8ebb8a..3bee7442 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,5 @@ nbformat >= 5.2.0 nbclient >= 0.2.0 tqdm >= 4.32.2 requests -entrypoints tenacity >= 5.0.2 aiohttp >=3.9.0; python_version=="3.12" From 9827118c2f75107c894d673b11b148f80580db52 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Sat, 11 Apr 2026 10:01:19 +0100 Subject: [PATCH 03/10] fix indent error Forwarded: no --- papermill/tests/test_iorw.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/papermill/tests/test_iorw.py b/papermill/tests/test_iorw.py index 7608dc42..dce5c0c9 100644 --- a/papermill/tests/test_iorw.py +++ b/papermill/tests/test_iorw.py @@ -108,7 +108,8 @@ def test_entrypoint_register(self): mock_entry_points = Mock() mock_entry_points.select.return_value = [fake_entrypoint] - with patch("papermill.iorw.entry_points", return_value=mock_entry_points): self.papermill_io.register_entry_points() + with patch("papermill.iorw.entry_points", return_value=mock_entry_points): + self.papermill_io.register_entry_points() mock_entry_points.select.assert_called_once_with(group="papermill.io") fake_ = self.papermill_io.get_handler("fake-from-entry-point://") assert fake_ == fake_entrypoint.load.return_value From b60779d8cf4153f04e18765834448c9839a57682 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Sat, 11 Apr 2026 10:31:30 +0100 Subject: [PATCH 04/10] Fix import error in test_autosave.py Forwarded: no --- papermill/tests/test_autosave.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/papermill/tests/test_autosave.py b/papermill/tests/test_autosave.py index 6a0b0bf9..dd3f80ae 100644 --- a/papermill/tests/test_autosave.py +++ b/papermill/tests/test_autosave.py @@ -10,6 +10,8 @@ from papermill.engines import NotebookExecutionManager from papermill.execute import execute_notebook +from . import get_notebook_path + class TestMidCellAutosave(unittest.TestCase): def setUp(self): From 6d6d1c49297d7b8fa45dfb9403d40d9f03c14860 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Sat, 11 Apr 2026 10:57:43 +0100 Subject: [PATCH 05/10] Fix NameError in test_execute.py Forwarded: no --- papermill/tests/test_execute.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/papermill/tests/test_execute.py b/papermill/tests/test_execute.py index c5abc95c..4acffd99 100644 --- a/papermill/tests/test_execute.py +++ b/papermill/tests/test_execute.py @@ -11,7 +11,7 @@ from nbformat import validate from papermill import engines, translators -from papermill.exceptions import PapermillExecutionError +from papermill.exceptions import PapermillExecutionError, strip_color from papermill.execute import execute_notebook from papermill.iorw import load_notebook_node from papermill.log import logger From 1dc1d2873b2a7e49680b4e4a5f4f162ede4b97c1 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Sat, 11 Apr 2026 11:34:28 +0100 Subject: [PATCH 06/10] Fix fake-engine name in test_engines.py Forwarded: no --- papermill/tests/test_engines.py | 1 + 1 file changed, 1 insertion(+) diff --git a/papermill/tests/test_engines.py b/papermill/tests/test_engines.py index a47f33de..8a4c9c55 100644 --- a/papermill/tests/test_engines.py +++ b/papermill/tests/test_engines.py @@ -490,6 +490,7 @@ def test_getting(self): def test_registering_entry_points(self): fake_entrypoint = Mock() + fake_entrypoint.name = "fake-engine" fake_entrypoint.load.return_value = Mock() mock_entry_points = Mock() From 23a6b8c2e51af1e5eedec5ac83279f09f3373091 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Sat, 11 Apr 2026 19:50:52 +0100 Subject: [PATCH 07/10] Set s3 bucket Location to eu-central-1 Forwarded: no --- papermill/tests/test_s3.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/papermill/tests/test_s3.py b/papermill/tests/test_s3.py index 8c6391a1..29c60581 100644 --- a/papermill/tests/test_s3.py +++ b/papermill/tests/test_s3.py @@ -168,7 +168,9 @@ def s3_client(): mock_aws.start() client = boto3.client('s3') - client.create_bucket(Bucket=test_bucket_name, CreateBucketConfiguration={'LocationConstraint': 'us-west-2'}) + client.create_bucket(Bucket=test_bucket_name, + CreateBucketConfiguration={'LocationConstraint': + 'eu-central-1'}) client.put_object(Bucket=test_bucket_name, Key=test_file_path, Body=test_nb_content) client.put_object(Bucket=test_bucket_name, Key=test_empty_file_path, Body='') yield S3() From 0d014ddf977215cb6047141a3556d207f3191755 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Fri, 17 Apr 2026 10:20:45 +0200 Subject: [PATCH 08/10] Patch pyproject.toml to limit pytest to papermill/tests Forwarded: no --- pyproject.toml | 1 + 1 file changed, 1 insertion(+) diff --git a/pyproject.toml b/pyproject.toml index 99336a0a..1d655570 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -197,6 +197,7 @@ quiet-level = 3 ignore-words-list = "dne, compiletime" [tool.pytest.ini_options] +testpaths = ["papermill/tests"] env = [ "AWS_SECRET_ACCESS_KEY=foobar_secret", "AWS_ACCESS_KEY_ID=foobar_key", From 6e9e59519e28e7c00d7533f5c4ee4b4b994b27a5 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 17 Apr 2026 10:34:43 +0000 Subject: [PATCH 09/10] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- papermill/tests/test_abs.py | 1 + papermill/tests/test_iorw.py | 2 +- papermill/tests/test_s3.py | 4 +--- pyproject.toml | 2 +- 4 files changed, 4 insertions(+), 5 deletions(-) diff --git a/papermill/tests/test_abs.py b/papermill/tests/test_abs.py index c383248d..cdbd0358 100644 --- a/papermill/tests/test_abs.py +++ b/papermill/tests/test_abs.py @@ -6,6 +6,7 @@ from papermill.abs import AzureBlobStore + class MockBytesIO: def __init__(self): self.list = [b"hello", b"world!"] diff --git a/papermill/tests/test_iorw.py b/papermill/tests/test_iorw.py index dce5c0c9..07d71754 100644 --- a/papermill/tests/test_iorw.py +++ b/papermill/tests/test_iorw.py @@ -102,7 +102,7 @@ def test_get_notebook_node_handler(self): def test_entrypoint_register(self): fake_entrypoint = Mock() - fake_entrypoint.name = "fake-from-entry-point://" + fake_entrypoint.name = "fake-from-entry-point://" fake_entrypoint.load.return_value = Mock() mock_entry_points = Mock() diff --git a/papermill/tests/test_s3.py b/papermill/tests/test_s3.py index 29c60581..34060765 100644 --- a/papermill/tests/test_s3.py +++ b/papermill/tests/test_s3.py @@ -168,9 +168,7 @@ def s3_client(): mock_aws.start() client = boto3.client('s3') - client.create_bucket(Bucket=test_bucket_name, - CreateBucketConfiguration={'LocationConstraint': - 'eu-central-1'}) + client.create_bucket(Bucket=test_bucket_name, CreateBucketConfiguration={'LocationConstraint': 'eu-central-1'}) client.put_object(Bucket=test_bucket_name, Key=test_file_path, Body=test_nb_content) client.put_object(Bucket=test_bucket_name, Key=test_empty_file_path, Body='') yield S3() diff --git a/pyproject.toml b/pyproject.toml index 1d655570..91d54c4f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -197,7 +197,7 @@ quiet-level = 3 ignore-words-list = "dne, compiletime" [tool.pytest.ini_options] -testpaths = ["papermill/tests"] +testpaths = [ "papermill/tests" ] env = [ "AWS_SECRET_ACCESS_KEY=foobar_secret", "AWS_ACCESS_KEY_ID=foobar_key", From 2954463e83f88f5975b702ecc86e036cd51eec00 Mon Sep 17 00:00:00 2001 From: Pieter Lenaerts Date: Mon, 25 May 2026 16:57:12 +0200 Subject: [PATCH 10/10] Fixing ruff errors --- papermill/engines.py | 2 +- papermill/iorw.py | 2 +- papermill/tests/test_cli.py | 1 + papermill/tests/test_clientwrap.py | 1 + papermill/tests/test_engines.py | 1 + papermill/tests/test_execute.py | 1 + papermill/tests/test_iorw.py | 1 + papermill/tests/test_parameterize.py | 2 +- 8 files changed, 8 insertions(+), 3 deletions(-) diff --git a/papermill/engines.py b/papermill/engines.py index c4659db2..89f93650 100644 --- a/papermill/engines.py +++ b/papermill/engines.py @@ -3,9 +3,9 @@ import datetime import sys from functools import wraps +from importlib.metadata import entry_points import dateutil -from importlib.metadata import entry_points from .clientwrap import PapermillNotebookClient from .exceptions import PapermillException diff --git a/papermill/iorw.py b/papermill/iorw.py index 8386e914..50f0f3c8 100644 --- a/papermill/iorw.py +++ b/papermill/iorw.py @@ -4,8 +4,8 @@ import sys import warnings from contextlib import contextmanager - from importlib.metadata import entry_points + import nbformat import requests import yaml diff --git a/papermill/tests/test_cli.py b/papermill/tests/test_cli.py index 095ede2b..c25161a5 100755 --- a/papermill/tests/test_cli.py +++ b/papermill/tests/test_cli.py @@ -17,6 +17,7 @@ from papermill import cli from papermill.cli import _is_float, _is_int, _resolve_type, papermill + from . import get_notebook_path, kernel_name diff --git a/papermill/tests/test_clientwrap.py b/papermill/tests/test_clientwrap.py index 6bf592f0..4409f2bd 100644 --- a/papermill/tests/test_clientwrap.py +++ b/papermill/tests/test_clientwrap.py @@ -6,6 +6,7 @@ from papermill.clientwrap import PapermillNotebookClient from papermill.engines import NotebookExecutionManager from papermill.log import logger + from . import get_notebook_path diff --git a/papermill/tests/test_engines.py b/papermill/tests/test_engines.py index 8a4c9c55..d6111b1b 100644 --- a/papermill/tests/test_engines.py +++ b/papermill/tests/test_engines.py @@ -10,6 +10,7 @@ from papermill.engines import Engine, NBClientEngine, NotebookExecutionManager from papermill.iorw import load_notebook_node from papermill.log import logger + from . import get_notebook_path diff --git a/papermill/tests/test_execute.py b/papermill/tests/test_execute.py index 4acffd99..d0584e11 100644 --- a/papermill/tests/test_execute.py +++ b/papermill/tests/test_execute.py @@ -16,6 +16,7 @@ from papermill.iorw import load_notebook_node from papermill.log import logger from papermill.utils import chdir + from . import get_notebook_path, kernel_name execute_notebook = partial(execute_notebook, kernel_name=kernel_name) diff --git a/papermill/tests/test_iorw.py b/papermill/tests/test_iorw.py index 07d71754..555cb00a 100644 --- a/papermill/tests/test_iorw.py +++ b/papermill/tests/test_iorw.py @@ -24,6 +24,7 @@ papermill_io, read_yaml_file, ) + from . import get_notebook_path FIXTURE_PATH = os.path.join(os.path.dirname(__file__), 'fixtures') diff --git a/papermill/tests/test_parameterize.py b/papermill/tests/test_parameterize.py index 71e77433..e1235600 100644 --- a/papermill/tests/test_parameterize.py +++ b/papermill/tests/test_parameterize.py @@ -1,10 +1,10 @@ import unittest from datetime import datetime - from papermill.exceptions import PapermillMissingParameterException from papermill.iorw import load_notebook_node from papermill.parameterize import add_builtin_parameters, parameterize_notebook, parameterize_path + from . import get_notebook_path