fix mypy

deepset-ai · Oct 23, 2024 · 2c2c0a2 · 2c2c0a2
1 parent 91eebae
commit 2c2c0a2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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