Skip to content
Snippets Groups Projects
Commit 05f08725 authored by JasonJiazhiZhang's avatar JasonJiazhiZhang Committed by Oleksandr
Browse files

Expose other measures through get_info in NavRLEnv (#120)

Expose other measures through get_info in NavRLEnv
parent 3b998621
No related branches found
No related tags found
No related merge requests found
......@@ -86,12 +86,7 @@ class NavRLEnv(habitat.RLEnv):
return done
def get_info(self, observations):
info = {}
if self.get_done(observations):
info["spl"] = self.habitat_env.get_metrics()["spl"]
return info
return self.habitat_env.get_metrics()
def make_env_fn(config_env, config_baseline, rank):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment