summaryrefslogtreecommitdiff
path: root/kafka/context.py
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/context.py')
-rw-r--r--kafka/context.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/kafka/context.py b/kafka/context.py
index 376fad1..d6c15fe 100644
--- a/kafka/context.py
+++ b/kafka/context.py
@@ -3,7 +3,8 @@ Context manager to commit/rollback consumer offsets.
"""
from logging import getLogger
-from kafka.common import check_error, OffsetCommitRequestPayload, OffsetOutOfRangeError
+from kafka.errors import check_error, OffsetOutOfRangeError
+from kafka.structs import OffsetCommitRequestPayload
class OffsetCommitContext(object):