def get_article_embeddings()

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

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

    def get_article_embeddings(self, article_ids, embedding_set=0):
        return self.embeddings[embedding_set][article_ids]