fix build (#15068)

airbytehq · Jul 27, 2022 · 2695c0c · 2695c0c
1 parent 8b763cc
commit 2695c0c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/airbyte-cdk/python/airbyte_cdk/sources/declarative/read_exception.py b/airbyte-cdk/python/airbyte_cdk/sources/declarative/read_exception.py
@@ -1,6 +1,8 @@
 #
 # Copyright (c) 2022 Airbyte, Inc., all rights reserved.
 #
+
+
 class ReadException(Exception):
     """
     Raise when there is an error reading data from an API Source

diff --git a/airbyte-cdk/python/unit_tests/sources/declarative/retrievers/test_simple_retriever.py b/airbyte-cdk/python/unit_tests/sources/declarative/retrievers/test_simple_retriever.py
@@ -8,6 +8,7 @@
 import pytest
 import requests
 from airbyte_cdk.models import SyncMode
+from airbyte_cdk.sources.declarative.read_exception import ReadException
 from airbyte_cdk.sources.declarative.requesters.error_handlers.response_action import ResponseAction
 from airbyte_cdk.sources.declarative.requesters.error_handlers.response_status import ResponseStatus
 from airbyte_cdk.sources.declarative.requesters.request_option import RequestOptionType
@@ -136,7 +137,7 @@ def test_parse_response(test_name, status_code, response_status, len_expected_re
         try:
             retriever.parse_response(response, stream_state={})
             assert False
-        except requests.exceptions.HTTPError:
+        except ReadException:
             pass
     else:
         records = retriever.parse_response(response, stream_state={})