def observation_space()

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


    def observation_space(self):
        # 2 embeddings (query and current page) plus the embeddings of
        # articles on the beam
        return spaces.Box(low=-np.inf, high=np.inf,
                          shape=(2 + self.beam_size, self.embedding_dim))