gameboy_worlds.interface.harvest_moon.registry

 1from typing import Dict, Type
 2from gameboy_worlds.interface.controller import Controller
 3from gameboy_worlds.interface.environment import Environment, DummyEnvironment
 4from gameboy_worlds.interface.harvest_moon.environments import (
 5    HarvestMoonEnvironment,
 6    HarvestMoonOCREnvironment,
 7    HarvestMoonTestEnvironment,
 8    HarvestMoonTrainEnvironment,
 9)
10from gameboy_worlds.interface.harvest_moon.controllers import HarvestMoonStateWiseController
11
12AVAILABLE_ENVIRONMENTS: Dict[str, Dict[str, Type[Environment]]] = {
13    "harvest_moon_1":{
14        "dummy": DummyEnvironment,
15        "default": HarvestMoonOCREnvironment,
16        "basic": HarvestMoonEnvironment,
17        "train": HarvestMoonTrainEnvironment,
18        "test": HarvestMoonTestEnvironment,
19    },
20    "harvest_moon_2":{
21        "default": HarvestMoonOCREnvironment,
22        "basic": HarvestMoonEnvironment,
23        "train": HarvestMoonTrainEnvironment,
24        "test": HarvestMoonTestEnvironment,
25    },
26    "harvest_moon_3":{
27        "default": HarvestMoonOCREnvironment,
28        "basic": HarvestMoonEnvironment,
29        "train": HarvestMoonTrainEnvironment,
30        "test": HarvestMoonTestEnvironment,
31    },
32}
33
34AVAILABLE_CONTROLLERS: Dict[str, Dict[str, Type[Controller]]] = {
35    "harvest_moon_1": {
36        "state_wise": HarvestMoonStateWiseController,
37    },
38    "harvest_moon_2": {
39        "state_wise": HarvestMoonStateWiseController,
40    },
41    "harvest_moon_3": {
42        "state_wise": HarvestMoonStateWiseController,
43    },
44}