from gym.envs.registration import register

register(
    id='powerworld-discrete-env-v0',
    entry_point='gym_powerworld.envs:DiscreteVoltageControlEnv',
)

register(
    id='powerworld-gridmind-env-v0',
    entry_point='gym_powerworld.envs:GridMindEnv',
)

register(
    id='powerworld-gridmind-hard-env-v0',
    entry_point='gym_powerworld.envs:GridMindHardEnv',
)

register(
    id='powerworld-gridmind-contingencies-env-v0',
    entry_point='gym_powerworld.envs:GridMindContingenciesEnv',
)

register(
    id='powerworld-discrete-env-simple-14-bus-v0',
    entry_point='gym_powerworld.envs:DiscreteVoltageControlSimple14BusEnv',
)

register(
    id='powerworld-discrete-env-gen-state-14-bus-v0',
    entry_point='gym_powerworld.envs:DiscreteVoltageControlGenState14BusEnv',
)

register(
    id='powerworld-discrete-env-branch-state-14-bus-v0',
    entry_point=
        'gym_powerworld.envs:DiscreteVoltageControlBranchState14BusEnv'
)

register(
    id='powerworld-discrete-env-branch-and-gen-state-14-bus-v0',
    entry_point=
        'gym_powerworld.envs:DiscreteVoltageControlBranchAndGenState14BusEnv'
)

register(
    id='powerworld-discrete-env-gen-shunt-no-contingencies-v0',
    entry_point=
        'gym_powerworld.envs:DiscreteVoltageControlGenAndShuntNoContingenciesEnv'
)

register(
    id='powerworld-discrete-env-gen-branch-shunt-v0',
    entry_point=
        'gym_powerworld.envs:DiscreteVoltageControlGensBranchesShuntsEnv'
)

register(
    id='powerworld-discrete-env-genvarfrac-branch-shunt-v0',
    entry_point=
        'gym_powerworld.envs:DiscreteVoltageControlGenVarFracBranchesShuntsEnv'
)