From 3e9420add1e74fb4e900e3cfee415e77343eae41 Mon Sep 17 00:00:00 2001 From: Marianne Stecklina Date: Thu, 19 Sep 2019 09:28:00 +0200 Subject: [PATCH] Make file reading more robust --- examples/utils_ner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/utils_ner.py b/examples/utils_ner.py index 0d3af3e061..39f6d08149 100644 --- a/examples/utils_ner.py +++ b/examples/utils_ner.py @@ -75,7 +75,7 @@ def read_examples_from_file(data_dir, evaluate=False): else: splits = line.split(" ") words.append(splits[0]) - labels.append(splits[-1][:-1]) + labels.append(splits[-1].replace("\n", "")) if words: examples.append(InputExample(guid="%s-%d".format(guid_prefix, guid_index), words=words,