#293 done. [related refactoring]

nicolay-r · Mar 4, 2022 · cd13621 · cd13621
1 parent eb6fd2e
commit cd13621
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 3 deletions.
diff --git a/arekit/processing/entities/base.py → examples/text/ner_base.py b/arekit/processing/entities/base.py → examples/text/ner_base.py
@@ -2,6 +2,9 @@
 
 
 class BaseNER(object):
+    """ CoNLL format based Named Entity Extractor
+        for list of input sequences, where sequence is a list of terms.
+    """
 
     separator = '-'
     begin_tag = 'B'

diff --git a/arekit/processing/entities/bert_ontonotes.py → examples/text/ner_ontonotes.py b/arekit/processing/entities/bert_ontonotes.py → examples/text/ner_ontonotes.py
@@ -1,5 +1,5 @@
 import importlib
-from arekit.processing.entities.base import BaseNER
+from examples.text.ner_base import BaseNER
 
 
 class BertOntonotesNER(BaseNER):

diff --git a/examples/text/pipeline_entities_bert_ontonotes.py b/examples/text/pipeline_entities_bert_ontonotes.py
@@ -2,8 +2,8 @@
 from arekit.common.entities.base import Entity
 from arekit.common.news.objects_parser import SentenceObjectsParserPipelineItem
 from arekit.common.text.partitioning.terms import TermsPartitioning
-from arekit.processing.entities.bert_ontonotes import BertOntonotesNER
 from arekit.processing.entities.obj_desc import NerObjectDescriptor
+from examples.text.ner_ontonotes import BertOntonotesNER
 
 
 class BertOntonotesNERPipelineItem(SentenceObjectsParserPipelineItem):

diff --git a/tests/processing/test_bert_ontonotes_ner.py → tests/examples/test_bert_ontonotes_ner.py b/tests/processing/test_bert_ontonotes_ner.py → tests/examples/test_bert_ontonotes_ner.py
@@ -1,7 +1,7 @@
 import unittest
 
-from arekit.processing.entities.bert_ontonotes import BertOntonotesNER
 from arekit.processing.entities.obj_desc import NerObjectDescriptor
+from examples.text.ner_ontonotes import BertOntonotesNER
 
 
 class BertOntonotesTest(unittest.TestCase):

diff --git a/.../test_bert_ontonotes_ner_pipeline_item.py → .../test_bert_ontonotes_ner_pipeline_item.py b/.../test_bert_ontonotes_ner_pipeline_item.py → .../test_bert_ontonotes_ner_pipeline_item.py