colosseum.mdp

 1"""
 2The module containing the MDP classes.
 3"""
 4
 5from typing import Union
 6
 7import numpy as np
 8
 9from colosseum.mdp.base import BaseMDP
10from colosseum.mdp.base_finite import EpisodicMDP
11from colosseum.mdp.base_infinite import ContinuousMDP
12from colosseum.mdp.custom_mdp import CustomNode as _CustomNode
13from colosseum.mdp.deep_sea.base import DeepSeaNode as _DeepSeaNode
14from colosseum.mdp.frozen_lake.base import FrozenLakeNode as _FrozenLakeNode
15from colosseum.mdp.minigrid_empty.base import MiniGridEmptyNode as _MiniGridEmptyNode
16from colosseum.mdp.minigrid_rooms.base import MiniGridRoomsNode as _NodeGridRooms
17from colosseum.mdp.river_swim.base import RiverSwimNode as _RiverSwimNode
18from colosseum.mdp.simple_grid.base import SimpleGridNode as _SimpleGridNode
19from colosseum.mdp.taxi.base import TaxiNode as _TaxiNode
20
21OBSERVATION_TYPE = Union[int, np.ndarray]
22ACTION_TYPE = Union[int, float, np.ndarray]
23REWARD_TYPE = Union[int, float, np.ndarray]
24
25NODE_TYPE = Union[
26    _CustomNode,
27    _RiverSwimNode,
28    _DeepSeaNode,
29    _FrozenLakeNode,
30    _SimpleGridNode,
31    _MiniGridEmptyNode,
32    _NodeGridRooms,
33    _TaxiNode,
34]