wassname/kair_algorithms_draft
1
0 Fork 0
mirror of https://github.com/wassname/kair_algorithms_draft.git synced 2026-10-29 13:10:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix episode unreset bug (#62)

Browse Source
This commit is contained in:
Whi Kwon authored and Donghyeon Kim committed 2019-05-30 22:39:44 +09:00
1 parent 47593785ad
commit 09b1f2e9a8
1 file changed
+3 -2
Download Patch File Download Diff File Expand all files Collapse all files
scripts/envs/open_manipulator/open_manipulator_reacher_env.py
+3 -2
View File
@@ -76,10 +76,11 @@ class OpenManipulatorReacherEnv(gym.Env):
if self.env_mode == "sim":
self.reward = self.compute_reward()
# TODO: Add termination condition
# if self.ros_interface.check_for_termination():
# self.done = True
# if self.ros_interface.check_for_termination():
# self.done = True
if self.ros_interface.check_for_success():
self.done = True
self.episode_steps = 0
obs = self.ros_interface.get_observation()