class Micro5_1Solver(BaseLearner):
    def __init__(self):
        self.current_map = {i:i for i in string.digits}
        self.question_mode = True

    def reward(self, r):
        print("qm:", self.question_mode, ", rew:", r)

    def next(self, input):
        if self.question_mode:
            self.question = input
            self.question_mode = False
            return self.current_map[input]

        self.current_map[self.question] = input
        self.question_mode = True
        return " "