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

Rename new envs (shorten Markovian to Markov)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-03-20 11:04:16 +01:00
parent 1936f8bf98
commit 9960201cb6
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fancy_gym/envs/__init__.py
View File

@ -316,18 +316,18 @@ register(
) )
register( register(
id='fancy/TableTennisMarkovian-v0', id='fancy/TableTennisMarkov-v0',
mp_wrapper=MPWrapper_TableTennis, 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, max_episode_steps=MAX_EPISODE_STEPS_TABLE_TENNIS_MARKOV_VER,
kwargs={ kwargs={
} }
) )
register( register(
id='fancy/TableTennisRndRobotMarkovian-v0', id='fancy/TableTennisRndRobotMarkov-v0',
mp_wrapper=MPWrapper_TableTennis_Rnd, 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, max_episode_steps=MAX_EPISODE_STEPS_TABLE_TENNIS_MARKOV_VER,
kwargs={ kwargs={
'random_pos_scale': 0.1, 'random_pos_scale': 0.1,

2
fancy_gym/envs/mujoco/table_tennis/table_tennis_env.py
View File

@ -299,7 +299,7 @@ class TableTennisEnv(MujocoEnv, utils.EzPickle):
return False, pos_traj, vel_traj return False, pos_traj, vel_traj
return True, pos_traj, vel_traj return True, pos_traj, vel_traj
class TableTennisMarkovian(TableTennisEnv): class TableTennisMarkov(TableTennisEnv):
def _get_reward2(self, hit_now, land_now): def _get_reward2(self, hit_now, land_now):
# Phase 1 not hit ball # Phase 1 not hit ball