Tianshou/tianshou/utils/cloudpicklewrapper.py
2020-03-11 09:09:56 +08:00

11 lines
260 B
Python

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)