from gym import envs

from config.metaworld_complex import complex_variant_config

envs.register(
    id="complex-v2",
    entry_point="diffgro.environments.metaworld_complex.envs:MetaWorldComplexEnv",
    kwargs={
        "max_steps": 2000,
    },
)

envs.register(
    id="complex-variant-v2",
    entry_point="diffgro.environments.metaworld_complex.envs:MetaWorldComplexVariantEnv",
    kwargs=complex_variant_config,  # Default config
)
