diff --git a/habitat_baselines/train_ppo.py b/habitat_baselines/train_ppo.py index d88edc3d229d193497909e7624c47d77974fc877..3e521a054da8dd30f78ebc9c99f62a64b2516ca3 100644 --- a/habitat_baselines/train_ppo.py +++ b/habitat_baselines/train_ppo.py @@ -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):