summaryrefslogtreecommitdiff
path: root/kafka/coordinator/assignors
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/coordinator/assignors')
-rw-r--r--kafka/coordinator/assignors/range.py2
-rw-r--r--kafka/coordinator/assignors/roundrobin.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/kafka/coordinator/assignors/range.py b/kafka/coordinator/assignors/range.py
index e4a7e33..861c8d2 100644
--- a/kafka/coordinator/assignors/range.py
+++ b/kafka/coordinator/assignors/range.py
@@ -1,7 +1,7 @@
import collections
import logging
-import six
+from kafka.vendor import six
from .abstract import AbstractPartitionAssignor
from ..protocol import ConsumerProtocolMemberMetadata, ConsumerProtocolMemberAssignment
diff --git a/kafka/coordinator/assignors/roundrobin.py b/kafka/coordinator/assignors/roundrobin.py
index 3fd3fd6..523a332 100644
--- a/kafka/coordinator/assignors/roundrobin.py
+++ b/kafka/coordinator/assignors/roundrobin.py
@@ -2,7 +2,7 @@ import collections
import itertools
import logging
-import six
+from kafka.vendor import six
from .abstract import AbstractPartitionAssignor
from ...common import TopicPartition