diff --git a/convlab2/util/multiwoz/lexicalize.py b/convlab2/util/multiwoz/lexicalize.py index d9ccaf5b8090456cdf489bbb0c98736f1b7f0ae5..14227b8c66745163e40c32876ea9ea13914a0a49 100755 --- a/convlab2/util/multiwoz/lexicalize.py +++ b/convlab2/util/multiwoz/lexicalize.py @@ -54,8 +54,9 @@ def lexicalize_da(meta, entities, state, requestable, cur_domain=None): else: continue for pair in v: - if pair[0] in state[domain.lower()]['semi']: - pair[1] = state[domain.lower()]['semi'][pair[0]] + slot = REF_SYS_DA[domain][pair[0]] + if slot in state[domain.lower()]['semi']: + pair[1] = state[domain.lower()]['semi'][slot] else: pair[1] = 'none' else: