def observe()

in gym3/interop.py [0:0]

• 12 lines of code
• 5 McCabe index (conditional complexity)

    def observe(self) -> Tuple[Any, Any, Any]:
        if self.have_new_action:
            ob, rew, first, self.info = self.bv_env.step_wait()
            self.have_new_action = False
            self.observe_tuple = (rew, ob, first)
            if self.render_mode == "rgb_array":
                renders = self.bv_env.get_images()
                for idx, img in enumerate(renders):
                    self.info[idx]["rgb"] = img
            elif self.render_mode is not None:
                self.bv_env.render(mode=self.render_mode)
        return self.observe_tuple