From 4f4290da593fed2400393e90c4c479a23e8ea003 Mon Sep 17 00:00:00 2001 From: zqwerty <zhuq96@hotmail.com> Date: Mon, 20 Dec 2021 13:24:21 +0000 Subject: [PATCH] add unified data loading function to convlab2/__init__.py --- convlab2/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/convlab2/__init__.py b/convlab2/__init__.py index 0fe7d5bf..a7b4c3c2 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 -- GitLab