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

Fix episode unreset bug (#62)

Browse Source
This commit is contained in:
Whi Kwon
2019-05-30 22:39:44 +09:00
committed by Donghyeon Kim
parent 47593785ad
commit 09b1f2e9a8
1 changed files with 3 additions and 2 deletions
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()