Add max_episode_steps to env

This commit is contained in:
whikwon
2019-04-15 19:27:50 +09:00
parent 771de4f98e
commit d7deb00c9b
2 changed files with 2 additions and 0 deletions
@@ -5,6 +5,7 @@ from geometry_msgs.msg import Quaternion
config = {
"ENV_NAME": "OpenManipulatorReacher",
"MAX_EPISODE_STEPS": 100,
"TERM_COUNT": 10,
"SUCCESS_COUNT": 10,
"OVERHEAD_ORIENTATION": Quaternion(
@@ -33,6 +33,7 @@ class OpenManipulatorReacherEnv(gym.Env):
self.cfg = cfg
self.env_name = self.cfg["ENV_NAME"]
self.env_mode = self.cfg["ENV_MODE"]
self._max_episode_steps = self.cfg["MAX_EPISODE_STEPS"]
self.reward_rescale_ratio = self.cfg["REWARD_RESCALE_RATIO"]
self.reward_func = self.cfg["REWARD_FUNC"]