From e6f48f654ec34d9cb29402889ba16a8dee4ebbc2 Mon Sep 17 00:00:00 2001 From: Suhail Patel Date: Mon, 3 Oct 2016 15:52:47 +0100 Subject: We only need to import cPickle explicitly if Python 2 --- pymemcache/serde.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/pymemcache/serde.py b/pymemcache/serde.py index 8c0a9da..8c88154 100644 --- a/pymemcache/serde.py +++ b/pymemcache/serde.py @@ -12,13 +12,17 @@ # See the License for the specific language governing permissions and # limitations under the License. -try: - import cPickle as pickle -except ImportError: - import pickle - import logging from io import BytesIO +import six + +if six.PY2: + try: + import cPickle as pickle + except ImportError: + import pickle +else: + import pickle try: long_type = long # noqa -- cgit v1.2.1