def step(self) -> None: super().step() for name in self.names: self.demonstration_protos[name] += proto_from_steps_and_action( self.step_result[name][0], self.step_result[name][1], self.action[name]) self.demonstration_protos[name] = self.demonstration_protos[name][ -self.n_demos:]
def step(self) -> None: super().step() for name in self.names: discrete_actions = (self.action[name].discrete if self.action_spec.discrete_size > 0 else None) continuous_actions = (self.action[name].continuous if self.action_spec.continuous_size > 0 else None) self.demonstration_protos[name] += proto_from_steps_and_action( self.step_result[name][0], self.step_result[name][1], continuous_actions, discrete_actions, ) self.demonstration_protos[name] = self.demonstration_protos[name][ -self.n_demos:]