def _observe()

in gym_wikinav/envs/wikinav_env/environment.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def _observe(self):
        if self._just_reset:
            self._query_embedding = \
                    self.graph.get_query_embeddings([self.navigator._path])[0]

        current_page_embedding = \
                self.graph.get_article_embeddings([self.cur_article_id])[0]
        beam_embeddings = self.graph.get_article_embeddings(self.navigator.beam)

        return self._query_embedding, current_page_embedding, beam_embeddings