From 222280e5b65aa2df7d6f91b45b20b713640cb5df Mon Sep 17 00:00:00 2001
From: zhuqi <zqwerty@users.noreply.github.com>
Date: Thu, 15 Oct 2020 21:22:58 +0800
Subject: [PATCH] Update policy_agenda_multiwoz.py
---
convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py b/convlab2/policy/rule/multiwoz/policy_agenda_multiwoz.py
index a13e5e8..97a8e33 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']
--
GitLab