update error message

huggingface · Jul 2, 2024 · 312c8b3 · 312c8b3
1 parent aaabdde
commit 312c8b3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
@@ -225,7 +225,8 @@ def _initialize_observation_space(self):
             )
         else:
             raise ValueError(
-                f"Unknown obs_type {self.obs_type}. Must be one of [pixels, state, pixels_agent_pos]"
+                f"Unknown obs_type {self.obs_type}. Must be one of [pixels, state, environment_state_agent_pos, "
+                "pixels_agent_pos]"
             )
 
     def _get_coverage(self):