From 688199fc34874f768e8beac51ad3c6cd45c474de Mon Sep 17 00:00:00 2001
From: Abhishek Kashyap <abhishek47kashyap@gmail.com>
Date: Thu, 5 May 2022 22:46:06 +0900
Subject: [PATCH] adds close()

---
 hsr_env.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/hsr_env.py b/hsr_env.py
index 1975f4f..b6c28ee 100644
--- a/hsr_env.py
+++ b/hsr_env.py
@@ -140,6 +140,9 @@ def __init__(self, connect=p.GUI):
 
         # print(self.robot.get_joint_infos())
 
+    def close(self):
+        p.disconnect(self.c_direct)
+
     def reset(self):
         self.c_gui.resetSimulation()
         self.c_direct.resetSimulation()