"TSocket read 0 bytes" error while running SQL query in a docker container

We are running a long-ish SQL query in pyhive from inside a python 3.6 container. The query freezes up after 10-15 mins and the error mentioned above is thrown. We are using the following versions :

thrift==0.13.0
sasl==0.2.1
thrift-sasl==0.4.2
PyHive==0.6.3

Any leads/solutions?

Additionally, when we try to query a different dedicated queue using configuration = {'mapred.job.queue.name' : queue_name} inside connect(), we also get the following error sometimes:

thrift.transport.TTransport.TTransportException: unexpected exception

Thanks!