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

13 lines
262 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)