Start OpenAI gym on arbitrary initial state

Question

Anybody knows any OpenAI Gym environments where we can set the initial state of the game? For example, I found the MountainCarContinuous-v0 can do such thing so that we can select at which point the car starts. However, I am looking for another more complex environment. Thanks in advance for your help!

Answer 1

You have to redefine the reset function of the class (for example, this). You may want to define it such that it gets as input your desired state, something like

def reset(self, state):
    self.state = state
    return np.array(self.state)

This should work for all OpenAI gym environments. If you want to do it for other simulators, things may be different. For instance, MuJoCo allows to do something like

saved_state = env.sim.get_state()
env.sim.set_state(saved_state)