From 4e589c9cbf8c228c967792a212a7e922354d6adf Mon Sep 17 00:00:00 2001 From: wangzy Date: Mon, 5 Feb 2024 14:22:52 +0800 Subject: [PATCH] fix default values of data item fields --- agent/streaming_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agent/streaming_inference.py b/agent/streaming_inference.py index 17033aa..922e057 100644 --- a/agent/streaming_inference.py +++ b/agent/streaming_inference.py @@ -518,7 +518,7 @@ def predict(args): d['idx'] = k d['query'] = d['question'].replace('$', '') d['gt'] = re.search('The answer is (.+)', d['answer'])[1] - d['pred'], d['steps'], d['error'] = None, [], [] + d['pred'], d['steps'], d['error'] = [], [], None return d dataset = load_dataset('gsm8k', 'main',