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

beerpong only one file

Browse Source
This commit is contained in:
Onur 2022-07-05 10:12:01 +02:00
parent 4dc33b0e97
commit 69de4286b3
4 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
alr_envs/alr/__init__.py
View File

@ -214,8 +214,8 @@ register(id='TableTennis4DCtxt-v0',
kwargs={'ctxt_dim': 4}) kwargs={'ctxt_dim': 4})
register( register(
id='ALRBeerPong-v0', id='BeerPong-v0',
entry_point='alr_envs.alr.mujoco:ALRBeerBongEnv', entry_point='alr_envs.alr.mujoco:BeerBongEnv',
max_episode_steps=300, max_episode_steps=300,
kwargs={ kwargs={
"frame_skip": 2 "frame_skip": 2
@ -223,8 +223,8 @@ register(
) )
register( register(
id='ALRBeerPong-v1', id='BeerPong-v1',
entry_point='alr_envs.alr.mujoco:ALRBeerBongEnv', entry_point='alr_envs.alr.mujoco:BeerBongEnv',
max_episode_steps=300, max_episode_steps=300,
kwargs={ kwargs={
"frame_skip": 2 "frame_skip": 2
@ -234,8 +234,8 @@ register(
# Here we use the same reward as in ALRBeerPong-v0, but now consider after the release, # Here we use the same reward as in ALRBeerPong-v0, but now consider after the release,
# only one time step, i.e. we simulate until the end of th episode # only one time step, i.e. we simulate until the end of th episode
register( register(
id='ALRBeerPongStepBased-v0', id='BeerPongStepBased-v0',
entry_point='alr_envs.alr.mujoco:ALRBeerBongEnvStepBased', entry_point='alr_envs.alr.mujoco:BeerBongEnvStepBased',
max_episode_steps=300, max_episode_steps=300,
kwargs={ kwargs={
"cup_goal_pos": [-0.3, -1.2], "cup_goal_pos": [-0.3, -1.2],
@ -245,8 +245,8 @@ register(
# Beerpong with episodic reward, but fixed release time step # Beerpong with episodic reward, but fixed release time step
register( register(
id='ALRBeerPongFixedRelease-v0', id='BeerPongFixedRelease-v0',
entry_point='alr_envs.alr.mujoco:ALRBeerBongEnvFixedReleaseStep', entry_point='alr_envs.alr.mujoco:BeerBongEnvFixedReleaseStep',
max_episode_steps=300, max_episode_steps=300,
kwargs={ kwargs={
"cup_goal_pos": [-0.3, -1.2], "cup_goal_pos": [-0.3, -1.2],
@ -414,7 +414,7 @@ for _v in _versions:
######################################################################################################################## ########################################################################################################################
## Beerpong ProMP ## Beerpong ProMP
_versions = ['ALRBeerPong-v0'] _versions = ['BeerPong-v0']
for _v in _versions: for _v in _versions:
_name = _v.split("-") _name = _v.split("-")
_env_id = f'{_name[0]}ProMP-{_name[1]}' _env_id = f'{_name[0]}ProMP-{_name[1]}'
@ -434,7 +434,7 @@ for _v in _versions:
ALL_ALR_MOTION_PRIMITIVE_ENVIRONMENTS["ProMP"].append(_env_id) ALL_ALR_MOTION_PRIMITIVE_ENVIRONMENTS["ProMP"].append(_env_id)
### BP with Fixed release ### BP with Fixed release
_versions = ["ALRBeerPongStepBased-v0", "ALRBeerPongFixedRelease-v0"] _versions = ["BeerPongStepBased-v0", "BeerPongFixedRelease-v0"]
for _v in _versions: for _v in _versions:
_name = _v.split("-") _name = _v.split("-")
_env_id = f'{_name[0]}ProMP-{_name[1]}' _env_id = f'{_name[0]}ProMP-{_name[1]}'

2
alr_envs/alr/mujoco/__init__.py
View File

@ -1,7 +1,7 @@
from .ant_jump.ant_jump import ALRAntJumpEnv from .ant_jump.ant_jump import ALRAntJumpEnv
from .ball_in_a_cup.ball_in_a_cup import ALRBallInACupEnv from .ball_in_a_cup.ball_in_a_cup import ALRBallInACupEnv
from .ball_in_a_cup.biac_pd import ALRBallInACupPDEnv from .ball_in_a_cup.biac_pd import ALRBallInACupPDEnv
from .beerpong.beerpong import BeerPongEnv from alr_envs.alr.mujoco.beerpong.beerpong import BeerPongEnv
from .half_cheetah_jump.half_cheetah_jump import ALRHalfCheetahJumpEnv from .half_cheetah_jump.half_cheetah_jump import ALRHalfCheetahJumpEnv
from .hopper_jump.hopper_jump_on_box import ALRHopperJumpOnBoxEnv from .hopper_jump.hopper_jump_on_box import ALRHopperJumpOnBoxEnv
from .hopper_throw.hopper_throw import ALRHopperThrowEnv from .hopper_throw.hopper_throw import ALRHopperThrowEnv

0
alr_envs/alr/mujoco/beerpong/deprecated/__init__.py Normal file
View File

0
alr_envs/alr/mujoco/beerpong/beerpong_reward_staged.py → alr_envs/alr/mujoco/beerpong/deprecated/beerpong_reward_staged.py
View File