from Environments.Chain import Chain
from Environments.FourRoomGridWorld import FourRoomGridWorld
environment_dict = {'FourRoomGridWorld': FourRoomGridWorld, 'Chain': Chain}