Fix retrying incomplete downloads with urllib3 2.x

s3transfer/__init__.py

@@ -135,7 +135,7 @@ def __call__(self, bytes_amount):
 import threading
 
 from botocore.compat import six  # noqa: F401
-from botocore.exceptions import IncompleteReadError
+from botocore.exceptions import IncompleteReadError, ResponseStreamingError
 from botocore.vendored.requests.packages.urllib3.exceptions import (
     ReadTimeoutError,
 )
@@ -624,6 +624,7 @@ def _download_range(
                     OSError,
                     ReadTimeoutError,
                     IncompleteReadError,
+                    ResponseStreamingError,
                 ) as e:
                     logger.debug(
                         "Retrying exception caught (%s), "
@@ -840,6 +841,7 @@ def _get_object(self, bucket, key, filename, extra_args, callback):
                 OSError,
                 ReadTimeoutError,
                 IncompleteReadError,
+                ResponseStreamingError,
             ) as e:
                 # TODO: we need a way to reset the callback if the
                 # download failed.

s3transfer/utils.py

@@ -21,7 +21,11 @@
 import threading
 from collections import defaultdict
 
-from botocore.exceptions import IncompleteReadError, ReadTimeoutError
+from botocore.exceptions import (
+    IncompleteReadError,
+    ReadTimeoutError,
+    ResponseStreamingError,
+)
 from botocore.httpchecksum import AwsChunkedWrapper
 from botocore.utils import is_s3express_bucket
 
@@ -41,6 +45,7 @@
     SOCKET_ERROR,
     ReadTimeoutError,
     IncompleteReadError,
+    ResponseStreamingError,
 )