from .q_learner import QLearner
from .qtran_learner import QLearner as QTranLearner
from .dtaps_learner import DTAPSLearner, ContDTAPSLearner

REGISTRY = {}
REGISTRY["q_learner"] = QLearner
REGISTRY["qtran_learner"] = QTranLearner
REGISTRY["dtaps_learner"] = DTAPSLearner
REGISTRY["cont_dtaps_learner"] = ContDTAPSLearner
