Tianshou/tianshou/utils/cloudpicklewrapper.py

11 lines
260 B
Python
Raw Normal View History

2020-03-11 09:09:56 +08:00
import cloudpickle
class CloudpickleWrapper(object):
def __init__(self, data):
self.data = data
def __getstate__(self):
return cloudpickle.dumps(self.data)
def __setstate__(self, data):
self.data = cloudpickle.loads(data)