test_action_wrappers.py 文件源码

def test_flattened_wrapper():
    expect = gym.make("ExpectTest-v0")
    md = spaces.MultiDiscrete([(0, 1), (0, 1)])
    expect.action_space = md
    expect.expectation  = (1, 1)
    wrapper = FlattenedActionWrapper(expect)
    wrapper.step(3)