EZPickle of a Meta-World env does not produce the exact same env

[reginald-mclean]

The arm position of the pickled env is incorrect for some reason

[pseudo-rnd-thoughts]

Could you provide a script to reproduce

[reginald-mclean]

import pickle
import numpy as np

mt1 = metaworld.MT1('assembly-v2', seed=5)
env = mt1.train_classes['assembly-v2']()
task = [task for task in mt1.train_tasks if task.env_name == 'assembly-v2'][0]
env.set_task(task)

pickled_env = pickle.loads(pickle.dumps(env))
obs1, _ = env.reset()
obs2, _ = pickled_env.reset()

assert np.all(np.equal(obs1, obs2))

[reginald-mclean]

This may also be fixed by #458

[reginald-mclean]

This seems to be an issue with where EZPickle is placed. Currently the init for EZPickle is placed in the Sawyer XYZ base class, but it may need to be moved into the individual classes for each environment.