def _generate_reset_input(self, training, config: ArenaConfig) -> UnityRLInput:
rl_in = UnityRLInput()
rl_in.is_training = training
rl_in.command = 1
rl_reset = UnityRLResetInput()
if (config is not None):
rl_reset.CopyFrom(config.dict_to_arena_config())
result = UnityInput()
result.rl_input.CopyFrom(rl_in)
result.rl_reset_input.CopyFrom(rl_reset)
return result
def wrap_unity_input(self, rl_input: UnityRLInput) -> UnityOutput:
result = UnityInput()
result.rl_input.CopyFrom(rl_input)
return result
def send_academy_parameters(
self, init_parameters: UnityRLInitializationInput
) -> UnityRLInitializationOutput:
inputs = UnityInput()
inputs.rl_initialization_input.CopyFrom(init_parameters)
return self.communicator.initialize(inputs).rl_initialization_output