use cloudpickle in save and load

Author: basnijholt

adaptive/learner/learnerND.py

@@ -3,8 +3,8 @@
 import random
 from collections import OrderedDict
 from collections.abc import Iterable
+from copy import deepcopy
 
-import cloudpickle
 import numpy as np
 import scipy.spatial
 from scipy import interpolate
@@ -1180,9 +1180,8 @@ def _get_plane_color(simplex):
         )
 
     def _get_data(self):
-        return cloudpickle.dumps(self.__dict__)
+        return deepcopy(self.__dict__)
 
     def _set_data(self, state):
-        state = cloudpickle.loads(state)
         for k, v in state.items():
             setattr(self, k, v)

adaptive/utils.py

@@ -6,6 +6,7 @@
 from contextlib import contextmanager
 from itertools import product
 
+import cloudpickle
 from atomicwrites import AtomicWriter
 
 
@@ -46,7 +47,7 @@ def save(fname, data, compress=True):
     if dirname:
         os.makedirs(dirname, exist_ok=True)
 
-    blob = pickle.dumps(data, protocol=pickle.HIGHEST_PROTOCOL)
+    blob = cloudpickle.dumps(data, protocol=pickle.HIGHEST_PROTOCOL)
     if compress:
         blob = gzip.compress(blob)
 
@@ -58,7 +59,7 @@ def load(fname, compress=True):
     fname = os.path.expanduser(fname)
     _open = gzip.open if compress else open
     with _open(fname, "rb") as f:
-        return pickle.load(f)
+        return cloudpickle.load(f)
 
 
 def copy_docstring_from(other):