diff options
-rw-r--r-- | src/matchbox/__init__.py | 5 | ||||
-rw-r--r-- | src/tinderbox/__init__.py | 5 |
2 files changed, 8 insertions, 2 deletions
diff --git a/src/matchbox/__init__.py b/src/matchbox/__init__.py index c6d470b..34c9daa 100644 --- a/src/matchbox/__init__.py +++ b/src/matchbox/__init__.py @@ -1,7 +1,10 @@ import socket import thread import protocol -import pickle +try: + import cPickle as pickle +except ImportError: + import pickle import StringIO import os import sys diff --git a/src/tinderbox/__init__.py b/src/tinderbox/__init__.py index 3d39493..eeb4647 100644 --- a/src/tinderbox/__init__.py +++ b/src/tinderbox/__init__.py @@ -1,7 +1,10 @@ import sys import protocol import socket -import pickle +try: + import cPickle as pickle +except ImportError: + import pickle from time import sleep import os import errno |