set obs dtype to float64

huggingface · Apr 5, 2024 · 6283bc7 · 6283bc7
1 parent 0fe4449
commit 6283bc7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
@@ -128,17 +128,17 @@ def __init__(
         if self.obs_type == "state":
             # [agent_x, agent_y, block_x, block_y, block_angle]
             self.observation_space = spaces.Box(
-                low=np.array([0, 0, 0, 0, 0], dtype=np.float32),
-                high=np.array([512, 512, 512, 512, 2 * np.pi], dtype=np.float32),
+                low=np.array([0, 0, 0, 0, 0], dtype=np.float64),
+                high=np.array([512, 512, 512, 512, 2 * np.pi], dtype=np.float64),
             )
         elif self.obs_type == "pixels":
             self.observation_space = spaces.Box(low=0, high=255, shape=(render_size, render_size, 3), dtype=np.uint8)
         elif self.obs_type == "pixels_agent_pos":
             self.observation_space = spaces.Dict({
                 "pixels": spaces.Box(low=0, high=255, shape=(render_size, render_size, 3), dtype=np.uint8),
                 "agent_pos": spaces.Box(
-                    low=np.array([0, 0], dtype=np.float32),
-                    high=np.array([512, 512], dtype=np.float32),
+                    low=np.array([0, 0], dtype=np.float64),
+                    high=np.array([512, 512], dtype=np.float64),
                 ),
             })