def __init__(self, env, width=<NUM_LIT>, height=<NUM_LIT>, grayscale=True):
gym.ObservationWrapper.__init__(self, env)<EOL>self.width = width<EOL>self.height = height<EOL>self.grayscale = grayscale<EOL>if self.grayscale:<EOL><INDENT>self.observation_space = spaces.Box(low=<NUM_LIT:0>, high=<NUM_LIT:255>,<EOL>shape=(self.height, self.width, <NUM_LIT:1>), dtype=np.uint8)<EOL><DEDENT>else:<EOL><INDENT>self.observation_space = spaces.Box(low=<NUM_LIT:0>, high=<NUM_LIT:255>,<EOL>shape=(self.height, self.width, <NUM_LIT:3>), dtype=np.uint8)<EOL><DEDENT>
Warp frames to 84x84 as done in the Nature paper and later work.