|
|
| import cloudpickle
|
|
|
|
|
| class PicklableWrapper:
|
| """
|
| Wrap an object to make it more picklable, note that it uses
|
| heavy weight serialization libraries that are slower than pickle.
|
| It's best to use it only on closures (which are usually not picklable).
|
|
|
| This is a simplified version of
|
| https://github.com/joblib/joblib/blob/master/joblib/externals/loky/cloudpickle_wrapper.py
|
| """
|
|
|
| def __init__(self, obj):
|
| while isinstance(obj, PicklableWrapper):
|
|
|
| obj = obj._obj
|
| self._obj = obj
|
|
|
| def __reduce__(self):
|
| s = cloudpickle.dumps(self._obj)
|
| return cloudpickle.loads, (s,)
|
|
|
| def __call__(self, *args, **kwargs):
|
| return self._obj(*args, **kwargs)
|
|
|
| def __getattr__(self, attr):
|
|
|
| if attr not in ["_obj"]:
|
| return getattr(self._obj, attr)
|
| return getattr(self, attr)
|
|
|
