Skip to content

Commit d55768e

Browse files
author
Marcin Kardas
committed
Extract tasks from paper's text
1 parent b0d0dae commit d55768e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sota_extractor2/models/linking/context_search.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -380,4 +380,5 @@ def from_paper(self, paper):
380380
return self(text)
381381

382382
def __call__(self, text):
383-
return find_datasets(normalize_cell_ws(normalize_dataset(text)))
383+
text = normalize_cell_ws(normalize_dataset(text))
384+
return find_datasets(text) | find_tasks(text)

0 commit comments

Comments
 (0)