def get_transition(self, frame_idx, env_idx):
past_frame, future_frame = self.get_frame_with_future(frame_idx, env_idx)<EOL>data_dict = {<EOL>'<STR_LIT>': past_frame,<EOL>'<STR_LIT>': future_frame,<EOL>'<STR_LIT>': self.action_buffer[frame_idx, env_idx],<EOL>'<STR_LIT>': self.reward_buffer[frame_idx, env_idx],<EOL>'<STR_LIT>': self.dones_buffer[frame_idx, env_idx],<EOL>}<EOL>for name in self.extra_data:<EOL><INDENT>data_dict[name] = self.extra_data[name][frame_idx, env_idx]<EOL><DEDENT>return data_dict<EOL>
Single transition with given index