#216 fixed

nicolay-r · Dec 10, 2021 · d376330 · d376330
1 parent 64301c4
commit d376330
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/pipeline.py b/examples/pipeline.py
@@ -11,6 +11,7 @@
 from arekit.common.text.options import TextParseOptions
 
 from arekit.contrib.experiment_rusentrel.common import entity_to_group_func
+from arekit.contrib.experiment_rusentrel.connotations.provider import RuSentiFramesConnotationProvider
 from arekit.contrib.experiment_rusentrel.entities.str_simple_fmt import StringEntitiesSimpleFormatter
 from arekit.contrib.experiment_rusentrel.labels.scalers.three import ThreeLabelScaler
 from arekit.contrib.experiment_rusentrel.synonyms.provider import RuSentRelSynonymsCollectionProvider
@@ -66,6 +67,8 @@ def extract(text):
     synonyms = RuSentRelSynonymsCollectionProvider.load_collection(
         stemmer=stemmer,
         version=RuSentRelVersions.V11)
+    frames_collection = RuSentiFramesCollection.read_collection(version=RuSentiFramesVersions.V20)
+    frames_connotation_provider = RuSentiFramesConnotationProvider(collection=frames_collection)
 
     labels_scaler = ThreeLabelScaler()
 
@@ -87,7 +90,7 @@ def extract(text):
                     binary=True),
                 string_entities_formatter=StringEntitiesSimpleFormatter(),
                 string_emb_entity_formatter=StringEntitiesSimpleFormatter())),
-        frames_collection=RuSentiFramesCollection.read_collection(version=RuSentiFramesVersions.V20),
+        frames_connotation_provider=frames_connotation_provider,
         frame_role_label_scaler=ThreeLabelScaler(),
         entity_to_group_func=entity_to_group_func,
         pos_terms_mapper=PosTermsMapper(None))

diff --git a/tests/contrib/source/test_ruattitudes.py b/tests/contrib/source/test_ruattitudes.py
@@ -3,12 +3,12 @@
 import unittest
 from tqdm import tqdm
 
-from arekit.common.text.options import TextParseOptions
 
 sys.path.append('../../../../')
 
 from arekit.common.opinions.base import Opinion
 from arekit.common.entities.base import Entity
+from arekit.common.text.options import TextParseOptions
 from arekit.common.utils import progress_bar_iter
 
 from arekit.contrib.source.ruattitudes.text_object import TextObject