diff --git a/parso/cache.py b/parso/cache.py index 1479275..d0465d0 100644 --- a/parso/cache.py +++ b/parso/cache.py @@ -4,11 +4,15 @@ import sys import hashlib import gc import shutil -import pickle import platform import errno import logging +try: + import cPickle as pickle +except: + import pickle + from parso._compatibility import FileNotFoundError LOG = logging.getLogger(__name__) diff --git a/parso/pgen2/grammar.py b/parso/pgen2/grammar.py index 82774e5..e5f2114 100644 --- a/parso/pgen2/grammar.py +++ b/parso/pgen2/grammar.py @@ -16,7 +16,10 @@ fallback token code OP, but the parser needs the actual token code. """ -import pickle +try: + import cPickle as pickle +except: + import pickle class Grammar(object):