From 142ae9b5352e53a66ca5b0b0a6fe9953114116a1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=A8=E6=B3=93?= Date: Thu, 23 Jun 2022 13:34:53 +0800 Subject: [PATCH] lint file --- .../space/fields/dst_processors.py | 33 +++++++++---------- 1 file changed, 16 insertions(+), 17 deletions(-) diff --git a/modelscope/preprocessors/space/fields/dst_processors.py b/modelscope/preprocessors/space/fields/dst_processors.py index 5eea7be3..0dcbd1fd 100644 --- a/modelscope/preprocessors/space/fields/dst_processors.py +++ b/modelscope/preprocessors/space/fields/dst_processors.py @@ -154,7 +154,8 @@ utter3 = { 'User-2': 'I am looking for an expensive indian restaurant in the area of centre.', 'System-2': - 'Might I recommend Saffron Brasserie? That is an expensive Indian restaurant in the center of town. I can book a table for you, if you like.', + 'Might I recommend Saffron Brasserie? That is an expensive Indian restaurant ' + 'in the center of town. I can book a table for you, if you like.', 'Dialog_Act-2': { 'Restaurant-Recommend': [['area', 'center of town'], ['food', 'Indian'], @@ -342,7 +343,8 @@ history_states3 = [{}, { 'leaveAt': '' } } - }, {}] + }, {} + ] class DSTProcessor(object): @@ -379,7 +381,8 @@ class DSTProcessor(object): def _convert_inputs_to_utterances(self, inputs: dict, history_states: list): - """This method is to generate the utterances with user, sys, dialog_acts and metadata, while metadata is from the history_states or the output from the inference pipline""" + """This method is to generate the utterances with user, sys, dialog_acts and metadata, + while metadata is from the history_states or the output from the inference pipline""" utterances = [] user_inputs = [] @@ -426,8 +429,8 @@ class DSTProcessor(object): if isinstance(item, dict): for a in item: aa = a.lower().split('-') - if aa[1] == 'inform' or aa[1] == 'recommend' or aa[ - 1] == 'select' or aa[1] == 'book': + if aa[1] == 'inform' or aa[1] == 'recommend' or \ + aa[1] == 'select' or aa[1] == 'book': for i in item[a]: s = i[0].lower() v = i[1].lower().strip() @@ -507,8 +510,8 @@ class multiwoz22Processor(DSTProcessor): if isinstance(acts[d][t]['dialog_act'], dict): for a in acts[d][t]['dialog_act']: aa = a.lower().split('-') - if aa[1] == 'inform' or aa[1] == 'recommend' or aa[ - 1] == 'select' or aa[1] == 'book': + if aa[1] == 'inform' or aa[1] == 'recommend' \ + or aa[1] == 'select' or aa[1] == 'book': for i in acts[d][t]['dialog_act'][a]: s = i[0].lower() v = i[1].lower().strip() @@ -762,8 +765,7 @@ class multiwoz22Processor(DSTProcessor): if delexicalize_sys_utts and is_sys_utt: inform_dict = {slot: 'none' for slot in slot_list} for slot in slot_list: - if (str(dialog_id), str(turn_itr), - slot) in sys_inform_dict: + if (str(dialog_id), str(turn_itr), slot) in sys_inform_dict: inform_dict[slot] = sys_inform_dict[(str(dialog_id), str(turn_itr), slot)] @@ -800,8 +802,7 @@ class multiwoz22Processor(DSTProcessor): if s in booked_slots: value_label = booked_slots[s] # Remember modified slots and entire dialog state - if cs in slot_list and cumulative_labels[ - cs] != value_label: + if cs in slot_list and cumulative_labels[cs] != value_label: modified_slots[cs] = value_label cumulative_labels[cs] = value_label @@ -901,8 +902,7 @@ class multiwoz22Processor(DSTProcessor): # since correct slot assignment can not be guaranteed anymore. if label_value_repetitions and slot in diag_seen_slots_dict: if class_type == 'copy_value' and list( - diag_seen_slots_value_dict.values()).count( - value_label) > 1: + diag_seen_slots_value_dict.values()).count(value_label) > 1: class_type = 'none' usr_utt_tok_label = [0 for _ in usr_utt_tok_label] @@ -934,11 +934,10 @@ class multiwoz22Processor(DSTProcessor): class_type_dict[slot] = 'none' referral_dict[slot] = 'none' if analyze: - if slot not in diag_seen_slots_dict or value_label != diag_seen_slots_value_dict[ - slot]: + if slot not in diag_seen_slots_dict or value_label != diag_seen_slots_value_dict[slot]: print('(%s): %s, ' % (slot, value_label), end='') - elif slot in diag_seen_slots_dict and class_type == diag_seen_slots_dict[ - slot] and class_type != 'copy_value' and class_type != 'inform': + elif slot in diag_seen_slots_dict and class_type == diag_seen_slots_dict[slot] \ + and class_type != 'copy_value' and class_type != 'inform': # If slot has seen before and its class type did not change, label this slot a not present, # assuming that the slot has not actually been mentioned in this turn. # Exceptions are copy_value and inform. If a seen slot has been tagged as copy_value or inform,