Skip to content

Commit f5a98c8

Browse files
committed
change model init type
1 parent cbce7dc commit f5a98c8

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

src/domain/transformers/ner.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,12 @@
1515

1616
logger = logging.getLogger(__name__)
1717

18-
ModelInitResult: TypeAlias = Tuple[torch.nn.Module, AutoTokenizer, Pipeline, bool]
18+
ModelInitResult: TypeAlias = Tuple[
19+
AutoModelForTokenClassification, AutoTokenizer, Pipeline, bool
20+
]
1921

2022

21-
class NERModel(BaseTransformerModel[AutoModelForTokenClassification]):
23+
class NERModel(BaseTransformerModel[AutoModelForTokenClassification]): # type: ignore
2224
"""
2325
Entity recognition model wrapper.
2426
"""

0 commit comments

Comments
 (0)