dream

1
from deeppavlov.core.common.registry import register
2
from deeppavlov.core.models.component import Component
3

4

5
@register("question_sign_checker")
6
class QuestionSignChecker(Component):
7
    """This class adds question sign if it is absent or replaces dot with question sign"""
8

9
    def __init__(self, **kwargs):
10
        pass
11

12
    def __call__(self, questions):
13
        questions_sanitized = []
14
        for question in questions:
15
            if not question.endswith("?"):
16
                if question.endswith("."):
17
                    question = question[:-1] + "?"
18
                else:
19
                    question += "?"
20
            questions_sanitized.append(question)
21
        return questions_sanitized
22