From 4936098b5ea49d30a93a8d44904e8d6866e10d10 Mon Sep 17 00:00:00 2001 From: Ting-Jun Wang Date: Mon, 6 Nov 2023 18:30:31 +0800 Subject: [PATCH] fix: change 'swap' to 'found' --- adversarial_summary.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/adversarial_summary.py b/adversarial_summary.py index b52cd8b..eb63d89 100644 --- a/adversarial_summary.py +++ b/adversarial_summary.py @@ -19,13 +19,13 @@ for file in ['train', 'val_unseen', 'val_seen', 'train_seen', 'test', 'val_train 'path': i['path'], 'heading': float(i['heading']), 'instructions': [ i['instruction'] ], - 'swap': [ True if i['found'] else False ], + 'found': [ i['found'] ], 'id': i['id'], 'objId': i['objId'] } else: result[instruction_id]['instructions'].append(i['instruction']) - result[instruction_id]['swap'].append( True if i['found'] else False) + result[instruction_id]['found'].append( i['found'] ) output = [] for k, item in result.items():