Add coverage info

huggingface · May 11, 2024 · bba1edb · bba1edb
1 parent 9e99323
commit bba1edb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
@@ -237,6 +237,7 @@ def step(self, action):
         observation = self.get_obs()
         info = self._get_info()
         info["is_success"] = is_success
+        info["coverage"] = coverage
 
         truncated = False
         return observation, reward, terminated, truncated, info