diff --git a/docker/client.py b/docker/client.py index 49e85dd4..3358b33a 100644 --- a/docker/client.py +++ b/docker/client.py @@ -17,6 +17,7 @@ import os import re import shlex import struct +from socket import socket as socket_wrapper import warnings import requests @@ -264,7 +265,7 @@ class Client(requests.Session): def _stream_helper(self, response): """Generator for data coming from a chunked-encoded HTTP response.""" - socket_fp = self._get_raw_response_socket(response) + socket_fp = socket_wrapper(_sock=self._get_raw_response_socket(response)) socket_fp.setblocking(1) socket = socket_fp.makefile() while True: