gameboy_worlds.interface.deja_vu.registry

 1from typing import Dict, Type
 2from gameboy_worlds.interface.controller import Controller
 3from gameboy_worlds.interface.deja_vu.controllers import DejaVuStateWiseController
 4from gameboy_worlds.interface.deja_vu.environments import (
 5    DejaVuEnvironment,
 6    DejaVuOCREnvironment,
 7    DejaVuTestEnvironment,
 8)
 9from gameboy_worlds.interface.environment import Environment, DummyEnvironment
10
11
12AVAILABLE_ENVIRONMENTS: Dict[str, Dict[str, Type[Environment]]] = {
13    "deja_vu_1": {
14        "dummy": DummyEnvironment,
15        "default": DejaVuOCREnvironment,
16        "basic": DejaVuEnvironment,
17        "test": DejaVuTestEnvironment,
18    },
19    "deja_vu_2": {
20        "dummy": DummyEnvironment,
21        "default": DejaVuOCREnvironment,
22        "basic": DejaVuEnvironment,
23        "test": DejaVuTestEnvironment,
24    },
25}
26
27AVAILABLE_CONTROLLERS: Dict[str, Dict[str, Type[Controller]]] = {
28    "state_wise": DejaVuStateWiseController,
29}