From eda5ba47f28e6dc9e4187af464910174e330fa99 Mon Sep 17 00:00:00 2001 From: Reggie McLean Date: Fri, 30 Aug 2024 13:05:31 -0400 Subject: [PATCH] env-names in env-dict --- metaworld/env_dict.py | 350 +++++++++++++++++++++--------------------- 1 file changed, 175 insertions(+), 175 deletions(-) diff --git a/metaworld/env_dict.py b/metaworld/env_dict.py index 4afa5e5e4..20724de90 100644 --- a/metaworld/env_dict.py +++ b/metaworld/env_dict.py @@ -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, } @@ -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", ] ) @@ -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) @@ -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 = { @@ -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 = {