Merge pull request #539 from ARISE-Initiative/fix-gym-demo

fix demo_gym_functionality rendering issues after first ep

robosuite/demos/demo_gym_functionality.py

@@ -57,5 +57,5 @@
             if terminated or truncated:
                 print("Episode finished after {} timesteps".format(t + 1))
                 observation, info = env.reset()
-                env.close()
                 break
+        env.close()

robosuite/wrappers/gym_wrapper.py

@@ -134,3 +134,9 @@ def compute_reward(self, achieved_goal, desired_goal, info):
         """
         # Dummy args used to mimic Wrapper interface
         return self.env.reward()
+
+    def close(self):
+        """
+        wrapper for calling underlying env close function
+        """
+        self.env.close()