From 43c243254aec4bcb6f977f4f512fb181ec7e986e Mon Sep 17 00:00:00 2001 From: erenup Date: Wed, 28 Aug 2019 16:03:17 +0800 Subject: [PATCH] avoid invalid labels of truth --- examples/single_model_scripts/utils_multiple_choice.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/single_model_scripts/utils_multiple_choice.py b/examples/single_model_scripts/utils_multiple_choice.py index d33477a30d..34505195ed 100644 --- a/examples/single_model_scripts/utils_multiple_choice.py +++ b/examples/single_model_scripts/utils_multiple_choice.py @@ -249,6 +249,7 @@ class ArcProcessor(DataProcessor): return int(truth) - 1 else: logger.info("truth ERROR!") + examples = [] three_choice = 0 four_choice = 0 @@ -267,6 +268,7 @@ class ArcProcessor(DataProcessor): continue four_choice += 1 truth = str(normalize(data_raw["answerKey"])) + assert truth is not None question_choices = data_raw["question"] question = question_choices["stem"] id = data_raw["id"]