From 40b2b21eeff82c606c4bc248653e457bd38cd607 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Wed, 17 Jun 2020 23:33:51 +0200 Subject: [PATCH] small bug fix --- spacy/gold/example.pyx | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/spacy/gold/example.pyx b/spacy/gold/example.pyx index eec2f8a86..8dfeffe98 100644 --- a/spacy/gold/example.pyx +++ b/spacy/gold/example.pyx @@ -243,16 +243,17 @@ def _fix_legacy_dict_data(example_dict): token_dict = example_dict.get("token_annotation", {}) doc_dict = example_dict.get("doc_annotation", {}) for key, value in example_dict.items(): - if key in ("token_annotation", "doc_annotation"): - pass - elif key == "ids": - pass - elif key in ("cats", "links") and value: - doc_dict[key] = value - elif key in ("ner", "entities") and value: - doc_dict["entities"] = value - else: - token_dict[key] = value + if value: + if key in ("token_annotation", "doc_annotation"): + pass + elif key == "ids": + pass + elif key in ("cats", "links"): + doc_dict[key] = value + elif key in ("ner", "entities"): + doc_dict["entities"] = value + else: + token_dict[key] = value # Remap keys remapping = { "words": "ORTH",