def get_filled_buffer_with_dones(frame_history=<NUM_LIT:1>):
observation_space = gym.spaces.Box(low=<NUM_LIT:0>, high=<NUM_LIT:255>, shape=(<NUM_LIT:2>, <NUM_LIT:2>, <NUM_LIT:1>), dtype=int)<EOL>action_space = gym.spaces.Discrete(<NUM_LIT:4>)<EOL>buffer = CircularVecEnvBufferBackend(<EOL><NUM_LIT:20>, num_envs=<NUM_LIT:2>, observation_space=observation_space, action_space=action_space, frame_history=frame_history<EOL>)<EOL>v1 = np.ones(<NUM_LIT:8>).reshape((<NUM_LIT:2>, <NUM_LIT:2>, <NUM_LIT:2>, <NUM_LIT:1>))<EOL>done_set = {<NUM_LIT:2>, <NUM_LIT:5>, <NUM_LIT:10>, <NUM_LIT>, <NUM_LIT>, <NUM_LIT>, <NUM_LIT>}<EOL>for i in range(<NUM_LIT:30>):<EOL><INDENT>item = v1.copy()<EOL>item[<NUM_LIT:0>] *= (i+<NUM_LIT:1>)<EOL>item[<NUM_LIT:1>] *= <NUM_LIT:10> * (i+<NUM_LIT:1>)<EOL>done_array = np.array([i in done_set, (i+<NUM_LIT:1>) in done_set], dtype=bool)<EOL>buffer.store_transition(item, <NUM_LIT:0>, float(i)/<NUM_LIT:2>, done_array)<EOL><DEDENT>return buffer<EOL>
Return simple preinitialized buffer with some done's in there