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))