summaryrefslogtreecommitdiff
path: root/test/test_failover_integration.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_failover_integration.py')
-rw-r--r--test/test_failover_integration.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py
index 1d835e2..5082d7c 100644
--- a/test/test_failover_integration.py
+++ b/test/test_failover_integration.py
@@ -4,10 +4,9 @@ import time
from . import unittest
-from kafka import KafkaClient, SimpleConsumer
+from kafka import KafkaClient, SimpleConsumer, KeyedProducer
from kafka.common import TopicAndPartition, FailedPayloadsError, ConnectionError
from kafka.producer.base import Producer
-from kafka.producer import KeyedProducer
from kafka.util import kafka_bytestring
from test.fixtures import ZookeeperFixture, KafkaFixture