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

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

@ -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):