From 33c79f31d2262afbce8328111e557eebe5b156ff Mon Sep 17 00:00:00 2001 From: Onur Date: Thu, 7 Apr 2022 10:24:47 +0200 Subject: [PATCH] remove unnecessary print --- alr_envs/alr/mujoco/table_tennis/tt_gym.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/alr_envs/alr/mujoco/table_tennis/tt_gym.py b/alr_envs/alr/mujoco/table_tennis/tt_gym.py index 7079a1b..33db936 100644 --- a/alr_envs/alr/mujoco/table_tennis/tt_gym.py +++ b/alr_envs/alr/mujoco/table_tennis/tt_gym.py @@ -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 -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 = "target_ball" TABLE_NAME = 'table_tennis_table' @@ -56,7 +56,6 @@ class TTEnvGym(MujocoEnv, utils.EzPickle): self.hit_ball = False self.ball_contact_after_hit = False self._ids_set = False - self.n_step = 0 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_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]), "ball_pos": np.copy(self.sim.data.qpos[7:])} self.n_step += 1 - print(self.n_step) return ob, reward, done, info # might add some information here .... def set_context(self, context):