From 47593785adf7146ae3aa877e1c22514a912090d2 Mon Sep 17 00:00:00 2001 From: Whi Kwon <30768596+whikwon@users.noreply.github.com> Date: Thu, 30 May 2019 22:01:43 +0900 Subject: [PATCH] Fix success_count not reset issue (#59) --- scripts/envs/open_manipulator/ros_interface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/envs/open_manipulator/ros_interface.py b/scripts/envs/open_manipulator/ros_interface.py index 1c681fc..e47c385 100755 --- a/scripts/envs/open_manipulator/ros_interface.py +++ b/scripts/envs/open_manipulator/ros_interface.py @@ -306,7 +306,6 @@ class OpenManipulatorRosBaseInterface(object): self.success_count += 1 if self.success_count == self.cfg["SUCCESS_COUNT"]: print ("Current episode succeeded") - self.success_count = 0 return True else: return False @@ -388,6 +387,7 @@ class OpenManipulatorRosGazeboInterface(OpenManipulatorRosBaseInterface): # self.delete_target_block() self.init_robot_pose() + self.success_count = 0 time.sleep(0.5) self.set_target_block(block_pose)