Guarantee the correct matching.

nicolay-r · Jun 15, 2022 · d754221 · d754221
1 parent f506fd3
commit d754221
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/arekit/common/news/parsed/providers/base.py b/arekit/common/news/parsed/providers/base.py
@@ -13,6 +13,9 @@ def Name(self):
 
     def init_parsed_news(self, parsed_news):
         assert(isinstance(parsed_news, ParsedNews))
-        self._doc_entities = [DocumentEntity(id_in_doc=doc_id, value=entity.Value,
-                                             e_type=entity.Type, group_index=entity.GroupIndex)
-                              for doc_id, entity in enumerate(parsed_news.iter_entities())]
+        self._doc_entities = {}
+        for doc_id, entity in enumerate(parsed_news.iter_entities()):
+            self._doc_entities[doc_id] = DocumentEntity(id_in_doc=doc_id,
+                                                        value=entity.Value,
+                                                        e_type=entity.Type,
+                                                        group_index=entity.GroupIndex)