diff --git a/convlab/evaluator/multiwoz_eval.py b/convlab/evaluator/multiwoz_eval.py index cb6c8feb73aea6e4481a51fa0eb2466a6a07d1c6..f300914eb9382de570fbc08fae55e048f0f07ffa 100755 --- a/convlab/evaluator/multiwoz_eval.py +++ b/convlab/evaluator/multiwoz_eval.py @@ -6,10 +6,10 @@ import numpy as np import pdb from copy import deepcopy -from data.unified_datasets.multiwoz21.preprocess import reverse_da, reverse_da_slot_name_map +# from data.unified_datasets.multiwoz21.preprocess import reverse_da, reverse_da_slot_name_map from convlab.util.multiwoz.multiwoz_slot_trans import REF_SYS_DA from convlab.evaluator.evaluator import Evaluator -from data.unified_datasets.multiwoz21.preprocess import reverse_da_slot_name_map +# from data.unified_datasets.multiwoz21.preprocess import reverse_da_slot_name_map from convlab.policy.rule.multiwoz.policy_agenda_multiwoz import unified_format, act_dict_to_flat_tuple from convlab.util.multiwoz.dbquery import Database from convlab.util import relative_import_module_from_unified_datasets @@ -28,6 +28,7 @@ REF_SYS_DA_M['taxi']['phone'] = 'phone' REF_SYS_DA_M['taxi']['car'] = 'car type' reverse_da = relative_import_module_from_unified_datasets('multiwoz21', 'preprocess.py', 'reverse_da') +reverse_da_slot_name_map = relative_import_module_from_unified_datasets('multiwoz21', 'preprocess.py', 'reverse_da_slot_name_map') requestable = \ diff --git a/convlab/policy/lava/multiwoz/lava.py b/convlab/policy/lava/multiwoz/lava.py index 76d177396c4d9a9d4c7e20f18ce652f6f8852ad5..ec07d36eda6f467706a6e6cb0c30a2ba3b23d37c 100755 --- a/convlab/policy/lava/multiwoz/lava.py +++ b/convlab/policy/lava/multiwoz/lava.py @@ -11,7 +11,9 @@ from convlab.policy import Policy from convlab.util.file_util import cached_path from convlab.util.multiwoz.state import default_state # from convlab.util.multiwoz.dbquery import Database -from data.unified_datasets.multiwoz21.database import Database +# from data.unified_datasets.multiwoz21.database import Database +from convlab.util import load_database +Database = load_database('multiwoz21') from copy import deepcopy import json import os