fix return type of the [_extract_ner] function
Browse files
README.md
CHANGED
|
@@ -78,7 +78,7 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification, pipelin
|
|
| 78 |
transformers_logger = logging.getLogger("transformers")
|
| 79 |
transformers_logger.setLevel(logging.WARNING)
|
| 80 |
|
| 81 |
-
def _extract_ner(sent: str, ner: pipeline) ->
|
| 82 |
grouped_ents = []
|
| 83 |
current_ent = {}
|
| 84 |
|
|
|
|
| 78 |
transformers_logger = logging.getLogger("transformers")
|
| 79 |
transformers_logger.setLevel(logging.WARNING)
|
| 80 |
|
| 81 |
+
def _extract_ner(sent: str, ner: pipeline) -> list:
|
| 82 |
grouped_ents = []
|
| 83 |
current_ent = {}
|
| 84 |
|