#507 fixed misinformation and bugs after refactoring

nicolay-r · Aug 6, 2023 · d7dd428 · d7dd428
1 parent 1f840bb
commit d7dd428
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/arekit/contrib/utils/pipelines/sources/nerel/doc_ops.py b/arekit/contrib/utils/pipelines/sources/nerel/doc_ops.py
@@ -1,21 +1,20 @@
 from arekit.common.experiment.api.ops_doc import DocumentOperations
-from arekit.contrib.source.nerel.io_utils import NerelVersions
 from arekit.contrib.source.nerel.reader import NerelDocReader
+from arekit.contrib.source.nerel.versions import NerelVersions
 
 
 class NERELDocOperation(DocumentOperations):
-    """ Document reader for the collection of the RuSentNE competition 2023.
-        For more details please follow the following repository:
-        github: https://github.com/dialogue-evaluation/RuSentNE-evaluation
+    """ A Russian dataset with nested named entities, relations, events and linked entities.
+        https://github.com/nerel-ds/NEREL
     """
 
     def __init__(self, filename_by_id, version):
         """ filename_ids: dict
                 Dictionary of {id: filename}, where
                     - id: int
                     - filename: str
-            version: SentiNerelVersions
-                Specify the appropriate version of teh SentiNEREL collection.
+            version: NerelVersions
+                Specify the appropriate version of teh NEREL collection.
         """
         assert(isinstance(filename_by_id, dict))
         assert(isinstance(version, NerelVersions))

diff --git a/arekit/contrib/utils/pipelines/sources/nerel/extract_text_relations.py b/arekit/contrib/utils/pipelines/sources/nerel/extract_text_relations.py
@@ -1,6 +1,7 @@
 from arekit.common.experiment.api.ops_doc import DocumentOperations
 from arekit.common.experiment.data_type import DataType
-from arekit.contrib.source.nerel.io_utils import NerelIOUtils, NerelVersions
+from arekit.contrib.source.nerel.io_utils import NerelIOUtils
+from arekit.contrib.source.nerel.versions import NerelVersions
 from arekit.contrib.utils.pipelines.sources.nerel.doc_ops import NERELDocOperation
 from arekit.contrib.utils.pipelines.sources.nerel.labels_fmt import NerelAnyLabelFormatter
 from arekit.contrib.utils.pipelines.text_opinion.annot.predefined import PredefinedTextOpinionAnnotator