c794fa99318462b5b04b77cfff27428d85cfb443,tensorforce/contrib/unreal_engine.py,UE4Environment,execute,#UE4Environment#Any#,126

Before Change

        self.protocol.send(message, self.socket)
        // wait for response (blocks)
        response = self.protocol.recv(self.socket)
        if "obs_dict" not in response:
            raise TensorForceError("Message without field "obs_dict" received!")
        self.last_observation = response["obs_dict"]  // cache last observation

        return response["obs_dict"], response.get("is_terminal", False), response.get("reward", 0.0)

    @cached_property
    def states(self):
        observation_space = {}

After Change

        r = response.pop("_reward", 0.0)
        is_terminal = response.pop("_is_terminal", False)

        obs = self.extract_observation(response)
        self.last_observation = obs  // cache last observation
        return obs, is_terminal, r

    @cached_property
    def states(self):
        observation_space = {}

Instances