make aiohttp ClientError retriable (#385)

fsspec · Apr 30, 2021 · b792f65 · b792f65
1 parent 60be1b9
commit b792f65
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/gcsfs/core.py b/gcsfs/core.py
@@ -29,6 +29,7 @@
 import weakref
 
 from requests.exceptions import RequestException, ProxyError
+from aiohttp.client_exceptions import ClientError
 from fsspec.asyn import sync_wrapper, sync, AsyncFileSystem
 from fsspec.utils import stringify_path, setup_logging
 from fsspec.implementations.http import get_client
@@ -533,7 +534,13 @@ async def _call(
                 )
                 self.validate_response(status, contents, path, headers)
                 break
-            except (HttpError, RequestException, GoogleAuthError, ChecksumError) as e:
+            except (
+                HttpError,
+                RequestException,
+                GoogleAuthError,
+                ChecksumError,
+                ClientError,
+            ) as e:
                 if (
                     isinstance(e, HttpError)
                     and e.code == 400

diff --git a/gcsfs/utils.py b/gcsfs/utils.py
@@ -1,5 +1,6 @@
 import requests.exceptions
 import google.auth.exceptions
+import aiohttp.client_exceptions
 
 
 class HttpError(Exception):
@@ -36,6 +37,7 @@ class ChecksumError(Exception):
     requests.exceptions.SSLError,
     requests.exceptions.ContentDecodingError,
     google.auth.exceptions.RefreshError,
+    aiohttp.client_exceptions.ClientError,
     ChecksumError,
 )