Skip to content

Commit

Permalink
#507 fixed parameter name
Browse files Browse the repository at this point in the history
  • Loading branch information
nicolay-r committed Aug 6, 2023
1 parent 5f966f2 commit 761fb5f
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,24 +9,24 @@
from arekit.contrib.utils.pipelines.text_opinion.filters.entity_based import EntityBasedTextOpinionFilter


def create_text_relation_extraction_pipeline(sentinerel_version,
def create_text_relation_extraction_pipeline(nerel_version,
text_parser,
label_formatter=NerelAnyLabelFormatter(),
terms_per_context=50,
doc_ops=None,
docs_limit=None,
entity_filter=None):
assert(isinstance(sentinerel_version, NerelVersions))
assert(isinstance(nerel_version, NerelVersions))
assert(isinstance(doc_ops, DocumentOperations) or doc_ops is None)

data_folding = None

if doc_ops is None:
# Default Initialization.
filenames_by_ids, data_folding = NerelIOUtils.read_dataset_split(version=sentinerel_version,
filenames_by_ids, data_folding = NerelIOUtils.read_dataset_split(version=nerel_version,
docs_limit=docs_limit)
doc_ops = NERELDocOperation(filename_by_id=filenames_by_ids,
version=sentinerel_version)
version=nerel_version)

text_opinion_filters = [
EntityBasedTextOpinionFilter(entity_filter=entity_filter),
Expand Down

0 comments on commit 761fb5f

Please sign in to comment.