Change BufferUnderflowError to ConnectionError in conn._read_bytes()

Both errors are handled the same way when raised and caught, so this makes sense.

1 files changed, 2 insertions, 3 deletions

diff --git a/kafka/client.py b/kafka/client.py
index 821904c..33c6d77 100644
--- a/kafka/client.py
+++ b/kafka/client.py
@@ -3,11 +3,10 @@ from collections import defaultdict
 from functools import partial
 from itertools import count
 import logging
-import socket
 import time
 
 from kafka.common import (
-    ErrorMapping, TopicAndPartition, BufferUnderflowError, ConnectionError,
+    ErrorMapping, TopicAndPartition, ConnectionError,
     FailedPayloadsException
 )
 from kafka.conn import KafkaConnection
@@ -175,7 +174,7 @@ class KafkaClient(object):
                     continue
                 try:
                     response = conn.recv(requestId)
-                except (ConnectionError, BufferUnderflowError), e:
+                except ConnectionError, e:
                     log.warning("Could not receive response to request [%s] "
                                 "from server %s: %s", request, conn, e)
                     failed = True