From 312c8b391a7f33014595cfece7867543c37a055f Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Tue, 2 Jul 2024 14:50:55 +0100 Subject: [PATCH] update error message --- gym_pusht/envs/pusht.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py index 692ea4f..6fc13e5 100644 --- 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):