def _handle_pickup(self, i, rewards, fwd_pos, fwd_cell):
    if fwd_cell:
        if fwd_cell.can_pickup():
            if (fwd_cell.index in [0, self.agents[i].index]):
                fwd_cell.cur_pos = np.array([(- 1), (- 1)])
                self.grid.set(*fwd_pos, None)
                self._reward(i, rewards, fwd_cell.reward)
def _handle_pickup(self, i, rewards, fwd_pos, fwd_cell):
    if fwd_cell:
        if fwd_cell.can_pickup():
            if (fwd_cell.index in [0, self.agents[i].index]):
                fwd_cell.cur_pos = np.array([(- 1), (- 1)])
                self.grid.set(*fwd_pos, None)
                self._reward(i, rewards, fwd_cell.reward)