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:]
