from diffgro.environments.variant import *


complex_variant_config = {
    "max_steps": 2000,
    "variant_space": VariantSpace(
        {
            "arm_speed": Categorical(a=[1.0]),
            "wind_xspeed": Categorical(a=[1.0]),
            "wind_yspeed": Categorical(a=[1.0]),
            "goal_resistance": VariantSpace(
                {
                    "handle": Categorical(a=[0]),
                    "button": Categorical(a=[0]),
                    "drawer": Categorical(a=[0]),
                    "lever": Categorical(a=[0]),
                    "door": Categorical(a=[0]),
                }
            ),
        }
    ),
}
