From dc0178ef172a1feddc3a4cbb03539aec8ae97133 Mon Sep 17 00:00:00 2001
From: seemingwang <seemingwang@users.noreply.github.com>
Date: Thu, 28 Oct 2021 13:23:00 +0800
Subject: [PATCH] fix MultiSlotDataGenerator error (#36773)

---
 python/paddle/distributed/fleet/data_generator/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/distributed/fleet/data_generator/__init__.py b/python/paddle/distributed/fleet/data_generator/__init__.py
index 230ada2abec..2288aca43f7 100644
--- a/python/paddle/distributed/fleet/data_generator/__init__.py
+++ b/python/paddle/distributed/fleet/data_generator/__init__.py
@@ -11,6 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 
-from .data_generator import DataGenerator  # noqa: F401
+from .data_generator import DataGenerator, MultiSlotDataGenerator  # noqa: F401
 
 __all__ = []
-- 
GitLab