Skip to content

Commit 5b1b602

Browse files
committed
fix: allow use tnm in solid_tumor matcher
1 parent a722b17 commit 5b1b602

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

edsnlp/pipes/ner/disorders/solid_tumor/solid_tumor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,9 +119,9 @@ def __init__(
119119
self.use_tnm = use_tnm
120120

121121
if use_tnm:
122-
from edsnlp.pipes.ner.tnm import TNM
122+
from edsnlp.pipes.ner.tnm.tnm import TNMMatcher
123123

124-
self.tnm = TNM(nlp, pattern=None, attr="TEXT")
124+
self.tnm = TNMMatcher(nlp=nlp, attr="TEXT")
125125

126126
def process_tnm(self, doc):
127127
spans = self.tnm.process(doc)

0 commit comments

Comments
 (0)