Change missing ConnectionError to KafkaConnectionError

88manpreet · 88manpreet · commit f23e1662e93b · 2018-07-31T18:30:54.000-07:00
diff --git a/kafka/client.py b/kafka/client.py
@@ -406,7 +406,7 @@ def failed_payloads(payloads):
         host, port, afi = get_ip_port_afi(broker.host)
         try:
             conn = self._get_conn(host, broker.port, afi, broker.nodeId)
-        except ConnectionError:
+        except KafkaConnectionError:
             failed_payloads(payloads)
 
         else:
