Rooba
/
genienlp
mirror of https://github.com/stanford-oval/genienlp.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

rm dangling space in woz

This commit is contained in:
Bryan Marcus McCann 2018-06-27 20:51:14 +00:00
parent 60d7705da1
commit 6e27de247f
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
text/torchtext/datasets/generic.py
View File

@ -942,13 +942,16 @@ class WOZ(CQA, data.Dataset):
delta_state['request'].append(slot[1]) delta_state['request'].append(slot[1])
current_state['request'].append(slot[1]) current_state['request'].append(slot[1])
previous_state = current_state previous_state = current_state
answer = ', '.join([f'{x[0]}: {x[1]}' for x in delta_state['inform']]) + ';' answer = ''
if len(delta_state['inform']) > 0:
answer = ', '.join([f'{x[0]}: {x[1]}' for x in delta_state['inform']])
answer += ';'
if len(delta_state['request']) > 0: if len(delta_state['request']) > 0:
answer += ' ' answer += ' '
answer += ', '.join(delta_state['request']) answer += ', '.join(delta_state['request'])
ex = {'context': ' '.join(context.split()), ex = {'context': ' '.join(context.split()),
'question': ' '.join(question.split()), 'lang': lang, 'question': ' '.join(question.split()), 'lang': lang,
'answer': answer if len(answer.strip()) > 1 else 'None', 'answer': answer if len(answer) > 1 else 'None',
'lang_dialogue_turn': f'{lang}_{di}_{ti}'} 'lang_dialogue_turn': f'{lang}_{di}_{ti}'}
split_file.write(json.dumps(ex)+'\n') split_file.write(json.dumps(ex)+'\n')