Fix cache on PyPy

cloudpipe · Nov 25, 2016 · e0c1130 · e0c1130
1 parent 020f477
commit e0c1130
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cloudpickle/cloudpickle.py b/cloudpickle/cloudpickle.py
@@ -320,10 +320,10 @@ def save_function_tuple(self, func):
         write(pickle.TUPLE)
         write(pickle.REDUCE)  # applies _fill_function on the tuple
 
-
-    _extract_code_globals_cache = (weakref.WeakKeyDictionary()
-                                   if sys.version_info >= (2, 7)
-                                   else {})
+    _extract_code_globals_cache = (
+        weakref.WeakKeyDictionary()
+        if sys.version_info >= (2, 7) and not hasattr(sys, "sys.pypy_version_info")
+        else {})
 
     @classmethod
     def extract_code_globals(cls, co):