text suppression test 1

import prodigy
from prodigy.util import log


class InfiniteTextStream:
    def __init__(self):
        self.i = 0

    def get_task(self):
        task = {
            "text": str(self.i),
            "meta": {
                "source": "Mock"
            }
        }
        self.i += 1
        return task


@prodigy.recipe('test')
def test():
    """
    A Prodigy test.
    """
    stream = InfiniteTextStream()

    def ask_questions(stream):
        while True:
            yield stream.get_task()

    def receive_answers(answers):
        print("Receiving answers")
        for answer in answers:
            print("ANSWER", answer)

    return {
        'view_id': 'blocks',
        'config': {
            'blocks': [
                {'view_id': "ner_manual"},
                {'view_id': "text"}
            ]
        },
        'dataset': 'infinite_stream_test',
        'stream': ask_questions(stream),
        'update': receive_answers,
    }