"""
"""

from models.base_model import Simulator

class MockSimulator(Simulator):

    def __init__(self, *args, **kwargs):
        pass

    def strip_stoptokens(self, tokens):
        return tokens

    def generate(self, history, stream=True):
        for text in ['hello', 'world']:
            yield text, [11]