Skip to content

Commit

Permalink
env-names in env-dict
Browse files Browse the repository at this point in the history
  • Loading branch information
@reginald-mclean
reginald-mclean committed Aug 30, 2024
1 parent 4c96658 commit eda5ba4
Showing 1 changed file with 175 additions and 175 deletions.
350 changes: 175 additions & 175 deletions metaworld/env_dict.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,56 +24,56 @@
)

ENV_CLS_MAP = {
"assembly-V3": envs.SawyerNutAssemblyEnvV3,
"basketball-V3": envs.SawyerBasketballEnvV3,
"bin-picking-V3": envs.SawyerBinPickingEnvV3,
"box-close-V3": envs.SawyerBoxCloseEnvV3,
"button-press-topdown-V3": envs.SawyerButtonPressTopdownEnvV3,
"button-press-topdown-wall-V3": envs.SawyerButtonPressTopdownWallEnvV3,
"button-press-V3": envs.SawyerButtonPressEnvV3,
"button-press-wall-V3": envs.SawyerButtonPressWallEnvV3,
"coffee-button-V3": envs.SawyerCoffeeButtonEnvV3,
"coffee-pull-V3": envs.SawyerCoffeePullEnvV3,
"coffee-push-V3": envs.SawyerCoffeePushEnvV3,
"dial-turn-V3": envs.SawyerDialTurnEnvV3,
"disassemble-V3": envs.SawyerNutDisassembleEnvV3,
"door-close-V3": envs.SawyerDoorCloseEnvV3,
"door-lock-V3": envs.SawyerDoorLockEnvV3,
"door-open-V3": envs.SawyerDoorEnvV3,
"door-unlock-V3": envs.SawyerDoorUnlockEnvV3,
"hand-insert-V3": envs.SawyerHandInsertEnvV3,
"drawer-close-V3": envs.SawyerDrawerCloseEnvV3,
"drawer-open-V3": envs.SawyerDrawerOpenEnvV3,
"faucet-open-V3": envs.SawyerFaucetOpenEnvV3,
"faucet-close-V3": envs.SawyerFaucetCloseEnvV3,
"hammer-V3": envs.SawyerHammerEnvV3,
"handle-press-side-V3": envs.SawyerHandlePressSideEnvV3,
"handle-press-V3": envs.SawyerHandlePressEnvV3,
"handle-pull-side-V3": envs.SawyerHandlePullSideEnvV3,
"handle-pull-V3": envs.SawyerHandlePullEnvV3,
"lever-pull-V3": envs.SawyerLeverPullEnvV3,
"peg-insert-side-V3": envs.SawyerPegInsertionSideEnvV3,
"pick-place-wall-V3": envs.SawyerPickPlaceWallEnvV3,
"pick-out-of-hole-V3": envs.SawyerPickOutOfHoleEnvV3,
"reach-V3": envs.SawyerReachEnvV3,
"push-back-V3": envs.SawyerPushBackEnvV3,
"push-V3": envs.SawyerPushEnvV3,
"pick-place-V3": envs.SawyerPickPlaceEnvV3,
"plate-slide-V3": envs.SawyerPlateSlideEnvV3,
"plate-slide-side-V3": envs.SawyerPlateSlideSideEnvV3,
"plate-slide-back-V3": envs.SawyerPlateSlideBackEnvV3,
"plate-slide-back-side-V3": envs.SawyerPlateSlideBackSideEnvV3,
"peg-unplug-side-V3": envs.SawyerPegUnplugSideEnvV3,
"soccer-V3": envs.SawyerSoccerEnvV3,
"stick-push-V3": envs.SawyerStickPushEnvV3,
"stick-pull-V3": envs.SawyerStickPullEnvV3,
"push-wall-V3": envs.SawyerPushWallEnvV3,
"reach-wall-V3": envs.SawyerReachWallEnvV3,
"shelf-place-V3": envs.SawyerShelfPlaceEnvV3,
"sweep-into-V3": envs.SawyerSweepIntoGoalEnvV3,
"sweep-V3": envs.SawyerSweepEnvV3,
"window-open-V3": envs.SawyerWindowOpenEnvV3,
"window-close-V3": envs.SawyerWindowCloseEnvV3,
"assembly-v3": envs.SawyerNutAssemblyEnvV3,
"basketball-v3": envs.SawyerBasketballEnvV3,
"bin-picking-v3": envs.SawyerBinPickingEnvV3,
"box-close-v3": envs.SawyerBoxCloseEnvV3,
"button-press-topdown-v3": envs.SawyerButtonPressTopdownEnvV3,
"button-press-topdown-wall-v3": envs.SawyerButtonPressTopdownWallEnvV3,
"button-press-v3": envs.SawyerButtonPressEnvV3,
"button-press-wall-v3": envs.SawyerButtonPressWallEnvV3,
"coffee-button-v3": envs.SawyerCoffeeButtonEnvV3,
"coffee-pull-v3": envs.SawyerCoffeePullEnvV3,
"coffee-push-v3": envs.SawyerCoffeePushEnvV3,
"dial-turn-v3": envs.SawyerDialTurnEnvV3,
"disassemble-v3": envs.SawyerNutDisassembleEnvV3,
"door-close-v3": envs.SawyerDoorCloseEnvV3,
"door-lock-v3": envs.SawyerDoorLockEnvV3,
"door-open-v3": envs.SawyerDoorEnvV3,
"door-unlock-v3": envs.SawyerDoorUnlockEnvV3,
"hand-insert-v3": envs.SawyerHandInsertEnvV3,
"drawer-close-v3": envs.SawyerDrawerCloseEnvV3,
"drawer-open-v3": envs.SawyerDrawerOpenEnvV3,
"faucet-open-v3": envs.SawyerFaucetOpenEnvV3,
"faucet-close-v3": envs.SawyerFaucetCloseEnvV3,
"hammer-v3": envs.SawyerHammerEnvV3,
"handle-press-side-v3": envs.SawyerHandlePressSideEnvV3,
"handle-press-v3": envs.SawyerHandlePressEnvV3,
"handle-pull-side-v3": envs.SawyerHandlePullSideEnvV3,
"handle-pull-v3": envs.SawyerHandlePullEnvV3,
"lever-pull-v3": envs.SawyerLeverPullEnvV3,
"peg-insert-side-v3": envs.SawyerPegInsertionSideEnvV3,
"pick-place-wall-v3": envs.SawyerPickPlaceWallEnvV3,
"pick-out-of-hole-v3": envs.SawyerPickOutOfHoleEnvV3,
"reach-v3": envs.SawyerReachEnvV3,
"push-back-v3": envs.SawyerPushBackEnvV3,
"push-v3": envs.SawyerPushEnvV3,
"pick-place-v3": envs.SawyerPickPlaceEnvV3,
"plate-slide-v3": envs.SawyerPlateSlideEnvV3,
"plate-slide-side-v3": envs.SawyerPlateSlideSideEnvV3,
"plate-slide-back-v3": envs.SawyerPlateSlideBackEnvV3,
"plate-slide-back-side-v3": envs.SawyerPlateSlideBackSideEnvV3,
"peg-unplug-side-v3": envs.SawyerPegUnplugSideEnvV3,
"soccer-v3": envs.SawyerSoccerEnvV3,
"stick-push-v3": envs.SawyerStickPushEnvV3,
"stick-pull-v3": envs.SawyerStickPullEnvV3,
"push-wall-v3": envs.SawyerPushWallEnvV3,
"reach-wall-v3": envs.SawyerReachWallEnvV3,
"shelf-place-v3": envs.SawyerShelfPlaceEnvV3,
"sweep-into-v3": envs.SawyerSweepIntoGoalEnvV3,
"sweep-v3": envs.SawyerSweepEnvV3,
"window-open-v3": envs.SawyerWindowOpenEnvV3,
"window-close-v3": envs.SawyerWindowCloseEnvV3,
}


Expand Down Expand Up @@ -214,56 +214,56 @@ def initialize(env, seed=None, render_mode=None):

ALL_V3_ENVIRONMENTS = _get_env_dict(
[
"assembly-V3",
"basketball-V3",
"bin-picking-V3",
"box-close-V3",
"button-press-topdown-V3",
"button-press-topdown-wall-V3",
"button-press-V3",
"button-press-wall-V3",
"coffee-button-V3",
"coffee-pull-V3",
"coffee-push-V3",
"dial-turn-V3",
"disassemble-V3",
"door-close-V3",
"door-lock-V3",
"door-open-V3",
"door-unlock-V3",
"hand-insert-V3",
"drawer-close-V3",
"drawer-open-V3",
"faucet-open-V3",
"faucet-close-V3",
"hammer-V3",
"handle-press-side-V3",
"handle-press-V3",
"handle-pull-side-V3",
"handle-pull-V3",
"lever-pull-V3",
"pick-place-wall-V3",
"pick-out-of-hole-V3",
"pick-place-V3",
"plate-slide-V3",
"plate-slide-side-V3",
"plate-slide-back-V3",
"plate-slide-back-side-V3",
"peg-insert-side-V3",
"peg-unplug-side-V3",
"soccer-V3",
"stick-push-V3",
"stick-pull-V3",
"push-V3",
"push-wall-V3",
"push-back-V3",
"reach-V3",
"reach-wall-V3",
"shelf-place-V3",
"sweep-into-V3",
"sweep-V3",
"window-open-V3",
"window-close-V3",
"assembly-v3",
"basketball-v3",
"bin-picking-v3",
"box-close-v3",
"button-press-topdown-v3",
"button-press-topdown-wall-v3",
"button-press-v3",
"button-press-wall-v3",
"coffee-button-v3",
"coffee-pull-v3",
"coffee-push-v3",
"dial-turn-v3",
"disassemble-v3",
"door-close-v3",
"door-lock-v3",
"door-open-v3",
"door-unlock-v3",
"hand-insert-v3",
"drawer-close-v3",
"drawer-open-v3",
"faucet-open-v3",
"faucet-close-v3",
"hammer-v3",
"handle-press-side-v3",
"handle-press-v3",
"handle-pull-side-v3",
"handle-pull-v3",
"lever-pull-v3",
"pick-place-wall-v3",
"pick-out-of-hole-v3",
"pick-place-v3",
"plate-slide-v3",
"plate-slide-side-v3",
"plate-slide-back-v3",
"plate-slide-back-side-v3",
"peg-insert-side-v3",
"peg-unplug-side-v3",
"soccer-v3",
"stick-push-v3",
"stick-pull-v3",
"push-v3",
"push-wall-v3",
"push-back-v3",
"reach-v3",
"reach-wall-v3",
"shelf-place-v3",
"sweep-into-v3",
"sweep-v3",
"window-open-v3",
"window-close-v3",
]
)

Expand All @@ -275,16 +275,16 @@ def initialize(env, seed=None, render_mode=None):

MT10_V3 = _get_env_dict(
[
"reach-V3",
"push-V3",
"pick-place-V3",
"door-open-V3",
"drawer-open-V3",
"drawer-close-V3",
"button-press-topdown-V3",
"peg-insert-side-V3",
"window-open-V3",
"window-close-V3",
"reach-v3",
"push-v3",
"pick-place-v3",
"door-open-v3",
"drawer-open-v3",
"drawer-close-v3",
"button-press-topdown-v3",
"peg-insert-side-v3",
"window-open-v3",
"window-close-v3",
]
)
MT10_V3_ARGS_KWARGS = _get_args_kwargs(ALL_V3_ENVIRONMENTS, MT10_V3)
Expand All @@ -301,23 +301,23 @@ def initialize(env, seed=None, render_mode=None):

ML10_V3 = _get_train_test_env_dict(
train_env_names=[
"reach-V3",
"push-V3",
"pick-place-V3",
"door-open-V3",
"drawer-close-V3",
"button-press-topdown-V3",
"peg-insert-side-V3",
"window-open-V3",
"sweep-V3",
"basketball-V3",
"reach-v3",
"push-v3",
"pick-place-v3",
"door-open-v3",
"drawer-close-v3",
"button-press-topdown-v3",
"peg-insert-side-v3",
"window-open-v3",
"sweep-v3",
"basketball-v3",
],
test_env_names=[
"drawer-open-V3",
"door-close-V3",
"shelf-place-V3",
"sweep-into-V3",
"lever-pull-V3",
"drawer-open-v3",
"door-close-v3",
"shelf-place-v3",
"sweep-into-v3",
"lever-pull-v3",
],
)
ML10_ARGS_KWARGS = {
Expand All @@ -327,58 +327,58 @@ def initialize(env, seed=None, render_mode=None):

ML45_V3 = _get_train_test_env_dict(
train_env_names=[
"assembly-V3",
"basketball-V3",
"button-press-topdown-V3",
"button-press-topdown-wall-V3",
"button-press-V3",
"button-press-wall-V3",
"coffee-button-V3",
"coffee-pull-V3",
"coffee-push-V3",
"dial-turn-V3",
"disassemble-V3",
"door-close-V3",
"door-open-V3",
"drawer-close-V3",
"drawer-open-V3",
"faucet-open-V3",
"faucet-close-V3",
"hammer-V3",
"handle-press-side-V3",
"handle-press-V3",
"handle-pull-side-V3",
"handle-pull-V3",
"lever-pull-V3",
"pick-place-wall-V3",
"pick-out-of-hole-V3",
"push-back-V3",
"pick-place-V3",
"plate-slide-V3",
"plate-slide-side-V3",
"plate-slide-back-V3",
"plate-slide-back-side-V3",
"peg-insert-side-V3",
"peg-unplug-side-V3",
"soccer-V3",
"stick-push-V3",
"stick-pull-V3",
"push-wall-V3",
"push-V3",
"reach-wall-V3",
"reach-V3",
"shelf-place-V3",
"sweep-into-V3",
"sweep-V3",
"window-open-V3",
"window-close-V3",
"assembly-v3",
"basketball-v3",
"button-press-topdown-v3",
"button-press-topdown-wall-v3",
"button-press-v3",
"button-press-wall-v3",
"coffee-button-v3",
"coffee-pull-v3",
"coffee-push-v3",
"dial-turn-v3",
"disassemble-v3",
"door-close-v3",
"door-open-v3",
"drawer-close-v3",
"drawer-open-v3",
"faucet-open-v3",
"faucet-close-v3",
"hammer-v3",
"handle-press-side-v3",
"handle-press-v3",
"handle-pull-side-v3",
"handle-pull-v3",
"lever-pull-v3",
"pick-place-wall-v3",
"pick-out-of-hole-v3",
"push-back-v3",
"pick-place-v3",
"plate-slide-v3",
"plate-slide-side-v3",
"plate-slide-back-v3",
"plate-slide-back-side-v3",
"peg-insert-side-v3",
"peg-unplug-side-v3",
"soccer-v3",
"stick-push-v3",
"stick-pull-v3",
"push-wall-v3",
"push-v3",
"reach-wall-v3",
"reach-v3",
"shelf-place-v3",
"sweep-into-v3",
"sweep-v3",
"window-open-v3",
"window-close-v3",
],
test_env_names=[
"bin-picking-V3",
"box-close-V3",
"hand-insert-V3",
"door-lock-V3",
"door-unlock-V3",
"bin-picking-v3",
"box-close-v3",
"hand-insert-v3",
"door-lock-v3",
"door-unlock-v3",
],
)
ML45_ARGS_KWARGS = {
Expand Down

0 comments on commit eda5ba4

Please sign in to comment.