fix bug caused by def render func (#472)

metadriverse · Jul 18, 2023 · cc8ca5f · cc8ca5f
1 parent 8615563
commit cc8ca5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metadrive/envs/base_env.py b/metadrive/envs/base_env.py
@@ -366,7 +366,7 @@ def render(self,
 
         if mode != "human" and self.config["image_observation"]:
             # fetch img from img stack to be make this func compatible with other render func in RL setting
-            return self.vehicle.observations.img_obs.get_image()
+            return self.observations[DEFAULT_AGENT].img_obs.get_image()
 
         if mode == "rgb_array":
             assert self.config["use_render"], "You should create a Panda3d window before rendering images!"