gameboy_worlds.interface.runes_of_virtue.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.runes_of_virtue.controllers import (
 6    RunesOfVirtueStateWiseController,
 7)
 8from gameboy_worlds.interface.runes_of_virtue.environments import (
 9    RunesOfVirtueEnvironment,
10    RunesOfVirtueOCREnvironment,
11    RunesOfVirtueTestEnvironment,
12    RunesOfVirtueTrainEnvironment,
13)
14
15
16AVAILABLE_ENVIRONMENTS: Dict[str, Dict[str, Type[Environment]]] = {
17    "runes_of_virtue_1": {
18        "dummy": DummyEnvironment,
19        "default": RunesOfVirtueOCREnvironment,
20        "basic": RunesOfVirtueEnvironment,
21        "ocr": RunesOfVirtueOCREnvironment,
22        "train": RunesOfVirtueTrainEnvironment,
23        "test": RunesOfVirtueTestEnvironment,
24    },
25    "runes_of_virtue_2": {
26        "dummy": DummyEnvironment,
27        "default": RunesOfVirtueOCREnvironment,
28        "basic": RunesOfVirtueEnvironment,
29        "ocr": RunesOfVirtueOCREnvironment,
30        "train": RunesOfVirtueTrainEnvironment,
31        "test": RunesOfVirtueTestEnvironment,
32    },
33}
34
35
36AVAILABLE_CONTROLLERS: Dict[str, Dict[str, Type[Controller]]] = {
37    "runes_of_virtue_1": {
38        "state_wise": RunesOfVirtueStateWiseController,
39    },
40    "runes_of_virtue_2": {
41        "state_wise": RunesOfVirtueStateWiseController,
42    },
43}