Merge pull request #3387 from flairNLP/multi_entity_hunflair

Added support for biomedical datasets with multiple entity types
flairNLP · Feb 12, 2024 · d55c0e9 · d55c0e9
2 parents 17e2895 + 8e9eeb9
commit d55c0e9
Show file tree

Hide file tree

Showing 6 changed files with 1,150 additions and 405 deletions.
diff --git a/flair/data.py b/flair/data.py
@@ -1787,10 +1787,10 @@ def __init__(
 
     def __str__(self) -> str:
         output = (
-            f"MultiCorpus: "  # type: ignore[arg-type]
-            f"{len(self.train) if self.train else 0} train + "
-            f"{len(self.dev) if self.dev else 0} dev + "
-            f"{len(self.test) if self.test else 0} test sentences\n - "
+            f"MultiCorpus: "
+            f"{_len_dataset(self.train) if self.train else 0} train + "
+            f"{_len_dataset(self.dev) if self.dev else 0} dev + "
+            f"{_len_dataset(self.test) if self.test else 0} test sentences\n - "
         )
         output += "\n - ".join([f"{type(corpus).__name__} {corpus!s} - {corpus.name}" for corpus in self.corpora])
         return output