from mdp_playground.spaces.discrete_extended import DiscreteExtended
from mdp_playground.spaces.box_extended import BoxExtended
from mdp_playground.spaces.multi_discrete_extended import MultiDiscreteExtended
from mdp_playground.spaces.image_multi_discrete import ImageMultiDiscrete

__all__ = ["BoxExtended", "DiscreteExtended", "MultiDiscreteExtended", "ImageMultiDiscrete"]
