Infra: introduce retry func warpper

fsspec · Sep 12, 2024 · 9adce4b · 9adce4b
1 parent 4280724
commit 9adce4b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tosfs/stability.py b/tosfs/stability.py
@@ -122,4 +122,6 @@ def _is_retryable_tos_server_exception(e: TosError) -> bool:
 
 
 def _is_retryable_tos_client_exception(e: TosError) -> bool:
-    return isinstance(e, TosClientError) and e.cause in TOS_CLIENT_RETRYABLE_EXCEPTIONS
+    return isinstance(e, TosClientError) and any(
+        isinstance(e.cause, excp) for excp in TOS_CLIENT_RETRYABLE_EXCEPTIONS
+    )