Update env.py

env.py

@@ -121,6 +121,7 @@ class RoboCasaEnv(RoboCasaGymEnv):
         self.unwrapped.sim._render_context_offscreen.gl_ctx.free()
         observation, info = super().reset(seed, **kwargs)
         self._task_description = self.env.get_ep_meta().get("lang", self.task)
+        print(f"[RoboCasaEnv] task_description: {self._task_description!r}")
         return self._format_raw_obs(observation), info
     
     def _format_raw_obs(self, raw_obs: dict):
@@ -155,7 +156,7 @@ class RoboCasaEnv(RoboCasaGymEnv):
             return frame
         import cv2
         frame = frame.copy()
-        cv2.putText(frame, self.task, (10, 30), cv2.FONT_HERSHEY_SIMPLEX,
+        cv2.putText(frame, self._task_description or self.task, (10, 30), cv2.FONT_HERSHEY_SIMPLEX,
                     0.8, (255, 255, 255), 2, cv2.LINE_AA)
         return frame