We only need to import cPickle explicitly if Python 2

1 files 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