summaryrefslogtreecommitdiff
path: root/numpy/oldnumeric/compat.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/oldnumeric/compat.py')
-rw-r--r--numpy/oldnumeric/compat.py20
1 files changed, 14 insertions, 6 deletions
diff --git a/numpy/oldnumeric/compat.py b/numpy/oldnumeric/compat.py
index 3e1d53d0e..607dd0b90 100644
--- a/numpy/oldnumeric/compat.py
+++ b/numpy/oldnumeric/compat.py
@@ -95,12 +95,20 @@ def _LoadArray(fp):
return m
import pickle, copy
-class Unpickler(pickle.Unpickler):
- def load_array(self):
- self.stack.append(_LoadArray(self))
-
- dispatch = copy.copy(pickle.Unpickler.dispatch)
- dispatch['A'] = load_array
+if sys.version_info[0] >= 3:
+ class Unpickler(pickle.Unpickler):
+ # XXX: should we implement this? It's not completely straightforward
+ # to do.
+ def __init__(self, *a, **kw):
+ raise NotImplementedError(
+ "numpy.oldnumeric.Unpickler is not supported on Python 3")
+else:
+ class Unpickler(pickle.Unpickler):
+ def load_array(self):
+ self.stack.append(_LoadArray(self))
+
+ dispatch = copy.copy(pickle.Unpickler.dispatch)
+ dispatch['A'] = load_array
class Pickler(pickle.Pickler):
def __init__(self, *args, **kwds):