from .halfcheetah_medium import halfcheetah_medium_args
from .hopper_medium import hopper_medium_args
from .walker2d_medium import walker2d_medium_args
from .halfcheetah_medium_expert import halfcheetah_medium_expert_args
from .hopper_medium_expert import hopper_medium_expert_args
from .walker2d_medium_expert import walker2d_medium_expert_args

from .pen_human import pen_human_args
from .door_human import door_human_args
from .hammer_human import hammer_human_args
from .pen_expert import pen_expert_args
from .door_expert import door_expert_args
from .hammer_expert import hammer_expert_args


loaded_args = {
    "halfcheetah-medium-v2": halfcheetah_medium_args,
    "hopper-medium-v2": hopper_medium_args,
    "walker2d-medium-v2": walker2d_medium_args,
    "halfcheetah-medium-expert-v2": halfcheetah_medium_expert_args,
    "hopper-medium-expert-v2": hopper_medium_expert_args,
    "walker2d-medium-expert-v2": walker2d_medium_expert_args,
    "pen-human-v1": pen_human_args,
    "door-human-v1": door_human_args,
    "hammer-human-v1": hammer_human_args,
    "pen-expert-v1": pen_expert_args,
    "door-expert-v1": door_expert_args,
    "hammer-expert-v1": hammer_expert_args,

}