diff --git a/convlab2/__init__.py b/convlab2/__init__.py index 0fe7d5bf01a7ac06fff476149427712f8c7c7ee4..a7b4c3c2d972cbb2dab7b091a099e8591b50895c 100755 --- a/convlab2/__init__.py +++ b/convlab2/__init__.py @@ -6,7 +6,8 @@ from convlab2.policy import Policy from convlab2.nlg import NLG from convlab2.dialog_agent import Agent, PipelineAgent from convlab2.dialog_agent import Session, BiSession, DealornotSession -from convlab2.util.unified_datasets_util import load_dataset, load_database +from convlab2.util.unified_datasets_util import load_dataset, load_database, load_unified_data, \ + load_nlu_data, load_dst_data, load_policy_data, load_nlg_data, load_e2e_data from os.path import abspath, dirname