def change_agent(self, fn): if self.last_t == -1: return agents_in_this_frame = self.timesteps[self.last_t] curr_adex = next((i for i,v in enumerate(agents_in_this_frame) if v==self.agent_num), 0) next_adex = util.cycle_index(curr_adex, fn, len(agents_in_this_frame)) self.agent_num = agents_in_this_frame[next_adex] self.reset_frame()
def change_agent(self, fn): if self.last_t == -1: return agents_in_this_frame = self.timesteps[self.last_t] curr_adex = next((i for i, v in enumerate(agents_in_this_frame) if v == self.agent_num), 0) next_adex = util.cycle_index(curr_adex, fn, len(agents_in_this_frame)) self.agent_num = agents_in_this_frame[next_adex] self.reset_frame()
def change_sample(self, fn): self.sample_num = util.cycle_index(self.sample_num, fn, util.NUM_SAMPLES) self.reset_frame()