mirror of
https://github.com/wassname/kair_algorithms_draft.git
synced 2026-09-10 12:14:41 +08:00
Add max_episode_steps to env
This commit is contained in:
@@ -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"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user