#337 fixed

nicolay-r · Jun 16, 2022 · 85f4c42 · 85f4c42
1 parent d754221
commit 85f4c42
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/arekit/common/news/parsed/providers/base.py b/arekit/common/news/parsed/providers/base.py
@@ -14,8 +14,9 @@ def Name(self):
     def init_parsed_news(self, parsed_news):
         assert(isinstance(parsed_news, ParsedNews))
         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)
+        for entity in parsed_news.iter_entities():
+            assert(entity.ID not in self._doc_entities)
+            self._doc_entities[entity.ID] = DocumentEntity(id_in_doc=entity.ID,
+                                                           value=entity.Value,
+                                                           e_type=entity.Type,
+                                                           group_index=entity.GroupIndex)