diff --git a/packages/ragbits-document-search/tests/unit/test_gcs_source.py b/packages/ragbits-document-search/tests/unit/test_gcs_source.py index 35d2a40c..da32b5a9 100644 --- a/packages/ragbits-document-search/tests/unit/test_gcs_source.py +++ b/packages/ragbits-document-search/tests/unit/test_gcs_source.py @@ -18,5 +18,5 @@ async def test_gcs_source_fetch(): assert path == TEST_FILE_PATH source = GCSSource(bucket="", object_name="not_found_file.py") - with pytest.raises(aiohttp.ClientResponseError, aiohttp.ClientConnectionError): + with pytest.raises(aiohttp.ClientConnectionError): await source.fetch()