fix bipedal rgb_array shape (openai#3083)

bojunliu0818 · Sep 11, 2022 · cae736b · cae736b
1 parent f4d668f
commit cae736b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gym/envs/box2d/bipedal_walker.py b/gym/envs/box2d/bipedal_walker.py
@@ -621,7 +621,9 @@ def render(self):
         if self.clock is None:
             self.clock = pygame.time.Clock()
 
-        self.surf = pygame.Surface((VIEWPORT_W + self.scroll * SCALE, VIEWPORT_H))
+        self.surf = pygame.Surface(
+            (VIEWPORT_W + max(0.0, self.scroll) * SCALE, VIEWPORT_H)
+        )
 
         pygame.transform.scale(self.surf, (SCALE, SCALE))