gameboy_worlds.interface.survival_kids.registry

 1from typing import Dict, Type
 2
 3from gameboy_worlds.interface.controller import Controller
 4from gameboy_worlds.interface.environment import DummyEnvironment, Environment
 5from gameboy_worlds.interface.survival_kids.environments import (
 6    SurvivalKidsEnvironment,
 7    SurvivalKidsOCREnvironment,
 8    SurvivalKidsTestEnvironment,
 9    SurvivalKidsTrainEnvironment,
10)
11
12
13AVAILABLE_ENVIRONMENTS: Dict[str, Dict[str, Type[Environment]]] = {
14    "survival_kids_1": {
15        "dummy": DummyEnvironment,
16        "default": SurvivalKidsOCREnvironment,
17        "basic": SurvivalKidsEnvironment,
18        "test": SurvivalKidsTestEnvironment,
19        "train": SurvivalKidsTrainEnvironment,
20    },
21    "survival_kids_2": {
22        "dummy": DummyEnvironment,
23        "default": SurvivalKidsOCREnvironment,
24        "basic": SurvivalKidsEnvironment,
25        "test": SurvivalKidsTestEnvironment,
26        "train": SurvivalKidsTrainEnvironment,
27    },
28}
29
30AVAILABLE_CONTROLLERS: Dict[str, Dict[str, Type[Controller]]] = {}