dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove unnecessary print

Browse Source
This commit is contained in:
Onur 2022-04-07 10:24:47 +02:00
parent 8ef00f7343
commit 33c79f31d2
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
alr_envs/alr/mujoco/table_tennis/tt_gym.py
View File

@ -10,7 +10,7 @@ from alr_envs.alr.mujoco.table_tennis.tt_reward import TT_Reward
#TODO: Check for simulation stability. Make sure the code runs even for sim crash #TODO: Check for simulation stability. Make sure the code runs even for sim crash
MAX_EPISODE_STEPS = 1375 # (1.25 + 1.5)/0.002 MAX_EPISODE_STEPS = 1375 # (1.25 + 1.5) /0.002
BALL_NAME_CONTACT = "target_ball_contact" BALL_NAME_CONTACT = "target_ball_contact"
BALL_NAME = "target_ball" BALL_NAME = "target_ball"
TABLE_NAME = 'table_tennis_table' TABLE_NAME = 'table_tennis_table'
@ -56,7 +56,6 @@ class TTEnvGym(MujocoEnv, utils.EzPickle):
self.hit_ball = False self.hit_ball = False
self.ball_contact_after_hit = False self.ball_contact_after_hit = False
self._ids_set = False self._ids_set = False
self.n_step = 0
super(TTEnvGym, self).__init__(model_path=model_path, frame_skip=1) super(TTEnvGym, self).__init__(model_path=model_path, frame_skip=1)
self.ball_id = self.sim.model._body_name2id[BALL_NAME] # find the proper -> not protected func. self.ball_id = self.sim.model._body_name2id[BALL_NAME] # find the proper -> not protected func.
self.ball_contact_id = self.sim.model._geom_name2id[BALL_NAME_CONTACT] self.ball_contact_id = self.sim.model._geom_name2id[BALL_NAME_CONTACT]
@ -168,7 +167,6 @@ class TTEnvGym(MujocoEnv, utils.EzPickle):
"q_pos": np.copy(self.sim.data.qpos[:7]), "q_pos": np.copy(self.sim.data.qpos[:7]),
"ball_pos": np.copy(self.sim.data.qpos[7:])} "ball_pos": np.copy(self.sim.data.qpos[7:])}
self.n_step += 1 self.n_step += 1
print(self.n_step)
return ob, reward, done, info # might add some information here .... return ob, reward, done, info # might add some information here ....
def set_context(self, context): def set_context(self, context):