def get_query_embeddings()

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

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

    def get_query_embeddings(self, paths, embedding_set=0):
        # Get the last non-STOP page in each corresponding path.
        last_pages = [[idx for idx in path if idx != self.stop_sentinel][-1]
                      for path in paths]
        return self.get_article_embeddings(last_pages,
                                           embedding_set=embedding_set)