diff --git a/convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py b/convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py index a13e5e8157a2fd0fc4a4ba0c1dc9f97b5d6f9993..97a8e33d08ff5cd26d8c9664954edbbc53e5c5e7 100755 --- a/convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py +++ b/convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py @@ -626,7 +626,7 @@ class Agenda(object): continue # For multiple choices, add new intent to select one: - if slot == 'choice' and value.strip().lower() not in ['0', 'zero']: + if slot == 'choice' and value.strip().lower() not in ['0', 'zero', '1', 'one']: self._push_item(domain + '-inform', "choice", "any") if slot in g_reqt: @@ -742,7 +742,7 @@ class Agenda(object): g_reqt, g_info, g_fail_info, g_book, g_fail_book = self._get_goal_infos(domain, goal) # delete Choice for slot, val in slot_vals: - if slot == 'choice': + if slot == 'choice' and val.strip().lower() not in ['0', 'zero', '1', 'one']: self._push_item(domain + '-inform', "choice", "any") slot_vals = [[slot, val] for [slot, val] in slot_vals if slot != 'choice']