Fix cursor position

huggingface · Apr 10, 2024 · c1322ad · c1322ad
1 parent 824b228
commit c1322ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gym_pusht/envs/pusht.py b/gym_pusht/envs/pusht.py
@@ -289,7 +289,7 @@ def _get_img(self, screen, width, height, render_action=False):
         render_size = min(width, height)
         if render_action and self._last_action is not None:
             action = np.array(self._last_action)
-            coord = (action / 512 * 96).astype(np.int32)
+            coord = (action / 512 * [height, width]).astype(np.int32)
             marker_size = int(8 / 96 * render_size)
             thickness = int(1 / 96 * render_size)
             cv2.drawMarker(