diff --git a/dlt/sources/helpers/requests/__init__.py b/dlt/sources/helpers/requests/__init__.py index d76e24ec42..3e29a2cf52 100644 --- a/dlt/sources/helpers/requests/__init__.py +++ b/dlt/sources/helpers/requests/__init__.py @@ -15,12 +15,11 @@ from requests.exceptions import ChunkedEncodingError from dlt.sources.helpers.requests.retry import Client from dlt.sources.helpers.requests.session import Session -from dlt.sources.helpers.rest_client import paginate from dlt.common.configuration.specs import RunConfiguration client = Client() -get, post, put, patch, delete, options, head, request, paginate = ( +get, post, put, patch, delete, options, head, request = ( client.get, client.post, client.put, @@ -29,7 +28,6 @@ client.options, client.head, client.request, - paginate, ) diff --git a/tests/sources/helpers/rest_client/test_requests_paginate.py b/tests/sources/helpers/rest_client/test_requests_paginate.py index 5ea137c735..43b2a412db 100644 --- a/tests/sources/helpers/rest_client/test_requests_paginate.py +++ b/tests/sources/helpers/rest_client/test_requests_paginate.py @@ -1,6 +1,6 @@ import pytest -from dlt.sources.helpers.requests import paginate +from dlt.sources.helpers.rest_client import paginate from dlt.sources.helpers.rest_client.paginators import JSONResponsePaginator from .conftest import assert_pagination