diff --git a/fancy_gym/envs/__init__.py b/fancy_gym/envs/__init__.py index a774d5d..38b5d34 100644 --- a/fancy_gym/envs/__init__.py +++ b/fancy_gym/envs/__init__.py @@ -316,18 +316,18 @@ register( ) register( - id='fancy/TableTennisMarkovian-v0', + id='fancy/TableTennisMarkov-v0', mp_wrapper=MPWrapper_TableTennis, - entry_point='fancy_gym.envs.mujoco:TableTennisMarkovian', + entry_point='fancy_gym.envs.mujoco:TableTennisMarkov', max_episode_steps=MAX_EPISODE_STEPS_TABLE_TENNIS_MARKOV_VER, kwargs={ } ) register( - id='fancy/TableTennisRndRobotMarkovian-v0', + id='fancy/TableTennisRndRobotMarkov-v0', mp_wrapper=MPWrapper_TableTennis_Rnd, - entry_point='fancy_gym.envs.mujoco:TableTennisMarkovian', + entry_point='fancy_gym.envs.mujoco:TableTennisMarkov', max_episode_steps=MAX_EPISODE_STEPS_TABLE_TENNIS_MARKOV_VER, kwargs={ 'random_pos_scale': 0.1, diff --git a/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py b/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py index 8155175..e6a14f8 100644 --- a/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py +++ b/fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py @@ -299,7 +299,7 @@ class TableTennisEnv(MujocoEnv, utils.EzPickle): return False, pos_traj, vel_traj return True, pos_traj, vel_traj -class TableTennisMarkovian(TableTennisEnv): +class TableTennisMarkov(TableTennisEnv): def _get_reward2(self, hit_now, land_now): # Phase 1 not hit ball