Commit 5d53f280 authored by Vít Starý Novotný's avatar Vít Starý Novotný
Browse files

Fix type error

parent 587af7d9

ahisto_named_entity_search/classification/model.py

+1 −0
Original line number Diff line number Diff line
@@ -18,6 +18,7 @@ class ClassificationModel: 
        is_text_from_index, = self.are_texts_from_index([text])



    def are_texts_from_index(self, texts: Iterable[TaggedSentence]) -> Iterable[Optional[bool]]:

        texts = list(texts)

        holes, samples = set(), []

        for text_number, text in enumerate(texts):

            sample = get_text_features(text)