Skip to content

Commit ecd793a

Browse files
committed
update deprecated character_extractor_kwargs argument for preconfigured pipelines
1 parent bbae55a commit ecd793a

1 file changed

Lines changed: 12 additions & 12 deletions

File tree

renard/pipeline/preconfigured.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ def co_occurrence_pipeline(
77
tokenizer_kwargs: Optional[dict] = None,
88
ner_step: Type[PipelineStep] = NLTKNamedEntityRecognizer,
99
ner_kwargs: Optional[dict] = None,
10-
characters_extractor_kwargs: Optional[dict] = None,
10+
character_unifier_kwargs: Optional[dict] = None,
1111
graph_extractor_kwargs: Optional[dict] = None,
1212
**pipeline_kwargs,
1313
) -> Pipeline:
@@ -16,7 +16,7 @@ def co_occurrence_pipeline(
1616
:param tokenizer_kwargs: kwargs for :class:`.NLTKTokenizer`
1717
:param ner_step: the class of the NER step to use.
1818
:param ner_kwargs: kwargs for :class:`.BertNamedEntityRecognizer`
19-
:param characters_extractor_kwargs: kwargs for
19+
:param character_unifier_kwargs: kwargs for
2020
:class:`.GraphRulesCharacterUnifier`
2121
:param graph_extractor_kwargs: kwargs for
2222
:class:`.CoOccurrencesGraphExtractor`,
@@ -30,7 +30,7 @@ def co_occurrence_pipeline(
3030

3131
tokenizer_kwargs = tokenizer_kwargs or {}
3232
ner_kwargs = ner_kwargs or {}
33-
characters_extractor_kwargs = characters_extractor_kwargs or {}
33+
character_unifier_kwargs = character_unifier_kwargs or {}
3434
graph_extractor_kwargs = graph_extractor_kwargs or {}
3535

3636
if not "co_occurrences_dist" in graph_extractor_kwargs:
@@ -40,7 +40,7 @@ def co_occurrence_pipeline(
4040
[
4141
NLTKTokenizer(**tokenizer_kwargs),
4242
ner_step(**ner_kwargs),
43-
GraphRulesCharacterUnifier(**characters_extractor_kwargs),
43+
GraphRulesCharacterUnifier(**character_unifier_kwargs),
4444
CoOccurrencesGraphExtractor(**graph_extractor_kwargs),
4545
],
4646
**pipeline_kwargs,
@@ -51,7 +51,7 @@ def conversational_pipeline(
5151
tokenizer_kwargs: Optional[dict] = None,
5252
ner_step: Type[PipelineStep] = NLTKNamedEntityRecognizer,
5353
ner_kwargs: Optional[dict] = None,
54-
characters_extractor_kwargs: Optional[dict] = None,
54+
character_unifier_kwargs: Optional[dict] = None,
5555
quote_detector_kwargs: Optional[dict] = None,
5656
speaker_detector_kwargs: Optional[dict] = None,
5757
graph_extractor_kwargs: Optional[dict] = None,
@@ -62,7 +62,7 @@ def conversational_pipeline(
6262
:param tokenizer_kwargs: kwargs for :class:`.NLTKTokenizer`
6363
:param ner_step: the type of the NER step to use.
6464
:param ner_kwargs: kwargs for :class:`.BertNamedEntityRecognizer`
65-
:param characters_extractor_kwargs: kwargs for
65+
:param character_unifier_kwargs: kwargs for
6666
:class:`.GraphRulesCharacterUnifier`
6767
:param quote_detector_kwargs: kwargs for :class:`.QuoteDetector`
6868
:param speaker_detector_kwargs: kwargs for
@@ -80,7 +80,7 @@ def conversational_pipeline(
8080

8181
tokenizer_kwargs = tokenizer_kwargs or {}
8282
ner_kwargs = ner_kwargs or {}
83-
characters_extractor_kwargs = characters_extractor_kwargs or {}
83+
character_unifier_kwargs = character_unifier_kwargs or {}
8484
quote_detector_kwargs = quote_detector_kwargs or {}
8585
speaker_detector_kwargs = speaker_detector_kwargs or {}
8686
graph_extractor_kwargs = graph_extractor_kwargs or {}
@@ -93,7 +93,7 @@ def conversational_pipeline(
9393
NLTKTokenizer(**tokenizer_kwargs),
9494
QuoteDetector(**quote_detector_kwargs),
9595
ner_step(**ner_kwargs),
96-
GraphRulesCharacterUnifier(**characters_extractor_kwargs),
96+
GraphRulesCharacterUnifier(**character_unifier_kwargs),
9797
BertSpeakerDetector(**speaker_detector_kwargs),
9898
ConversationalGraphExtractor(**graph_extractor_kwargs),
9999
],
@@ -105,7 +105,7 @@ def relational_pipeline(
105105
tokenizer_kwargs: Optional[dict] = None,
106106
ner_step: Type[PipelineStep] = NLTKNamedEntityRecognizer,
107107
ner_kwargs: Optional[dict] = None,
108-
characters_extractor_kwargs: Optional[dict] = None,
108+
character_unifier_kwargs: Optional[dict] = None,
109109
relation_extractor_kwargs: Optional[dict] = None,
110110
graph_extractor_kwargs: Optional[dict] = None,
111111
**pipeline_kwargs,
@@ -115,7 +115,7 @@ def relational_pipeline(
115115
:param tokenizer_kwargs: kwargs for :class:`.NLTKTokenizer`
116116
:param ner_step: the type of the NER step to use.
117117
:param ner_kwargs: kwargs for :class:`.BertNamedEntityRecognizer`
118-
:param characters_extractor_kwargs: kwargs for
118+
:param characters_unifier_kwargs: kwargs for
119119
:class:`.GraphRulesCharacterUnifier`
120120
:param relation_extraction_kwargs: kwargs for :class:`.T5RelationExtractor`
121121
:param pipeline_kwargs: kwargs for :class:`.Pipeline`
@@ -127,15 +127,15 @@ def relational_pipeline(
127127

128128
tokenizer_kwargs = tokenizer_kwargs or {}
129129
ner_kwargs = ner_kwargs or {}
130-
characters_extractor_kwargs = characters_extractor_kwargs or {}
130+
character_unifier_kwargs = character_unifier_kwargs or {}
131131
relation_extractor_kwargs = relation_extractor_kwargs or {}
132132
graph_extractor_kwargs = graph_extractor_kwargs or {}
133133

134134
return Pipeline(
135135
[
136136
NLTKTokenizer(**tokenizer_kwargs),
137137
ner_step(**ner_kwargs),
138-
GraphRulesCharacterUnifier(**characters_extractor_kwargs),
138+
GraphRulesCharacterUnifier(**character_unifier_kwargs),
139139
T5RelationExtractor(**relation_extractor_kwargs),
140140
RelationalGraphExtractor(**graph_extractor_kwargs),
141141
],

0 commit comments

Comments
 (0)