diff options
| -rw-r--r-- | kafka/protocol/pickle.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/kafka/protocol/pickle.py b/kafka/protocol/pickle.py index af0d1ee..2265efd 100644 --- a/kafka/protocol/pickle.py +++ b/kafka/protocol/pickle.py @@ -1,6 +1,10 @@ from __future__ import absolute_import -import copy_reg +try: + import copyreg +except ImportError: + import copy_reg as copyreg # python2 + import types @@ -22,4 +26,4 @@ def _unpickle_method(func_name, obj, cls): return func.__get__(obj, cls) # https://bytes.com/topic/python/answers/552476-why-cant-you-pickle-instancemethods -copy_reg.pickle(types.MethodType, _pickle_method, _unpickle_method) +copyreg.pickle(types.MethodType, _pickle_method, _unpickle_method) |