From 2c2c0a2a187cba39d6dc0bc606d933c4bb1eff50 Mon Sep 17 00:00:00 2001 From: tstadel Date: Wed, 23 Oct 2024 17:16:58 +0200 Subject: [PATCH] fix mypy --- deepset_cloud_sdk/_s3/upload.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deepset_cloud_sdk/_s3/upload.py b/deepset_cloud_sdk/_s3/upload.py index 529f281c..52b63d0a 100644 --- a/deepset_cloud_sdk/_s3/upload.py +++ b/deepset_cloud_sdk/_s3/upload.py @@ -27,7 +27,9 @@ class RetryableHttpError(Exception): """An error that indicates a function should be retried.""" - def __init__(self, error: Union[aiohttp.ClientResponseError, aiohttp.ServerDisconnectedError]) -> None: + def __init__( + self, error: Union[aiohttp.ClientResponseError, aiohttp.ServerDisconnectedError, aiohttp.ClientConnectionError] + ) -> None: """Store the original exception.""" self.error = error