Skip to content

Commit 091e329

Browse files
committed
update T5RelationExtractor to GenerativeRelationExtractor in preconfigured module
1 parent b2d65c0 commit 091e329

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

renard/pipeline/preconfigured.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -117,12 +117,13 @@ def relational_pipeline(
117117
:param ner_kwargs: kwargs for :class:`.BertNamedEntityRecognizer`
118118
:param characters_unifier_kwargs: kwargs for
119119
:class:`.GraphRulesCharacterUnifier`
120-
:param relation_extraction_kwargs: kwargs for :class:`.T5RelationExtractor`
120+
:param relation_extraction_kwargs: kwargs for
121+
:class:`.GenerativeRelationExtractor`
121122
:param pipeline_kwargs: kwargs for :class:`.Pipeline`
122123
"""
123124
from renard.pipeline.tokenization import NLTKTokenizer
124125
from renard.pipeline.character_unification import GraphRulesCharacterUnifier
125-
from renard.pipeline.relation_extraction import T5RelationExtractor
126+
from renard.pipeline.relation_extraction import GenerativeRelationExtractor
126127
from renard.pipeline.graph_extraction import RelationalGraphExtractor
127128

128129
tokenizer_kwargs = tokenizer_kwargs or {}
@@ -136,7 +137,7 @@ def relational_pipeline(
136137
NLTKTokenizer(**tokenizer_kwargs),
137138
ner_step(**ner_kwargs),
138139
GraphRulesCharacterUnifier(**character_unifier_kwargs),
139-
T5RelationExtractor(**relation_extractor_kwargs),
140+
GenerativeRelationExtractor(**relation_extractor_kwargs),
140141
RelationalGraphExtractor(**graph_extractor_kwargs),
141142
],
142143
**pipeline_kwargs,

0 commit comments

Comments
 (0)