From 35db3d17ec702e6cc55cfd6b56c19a03c801e391 Mon Sep 17 00:00:00 2001
From: Xing Wu <wuxing03@baidu.com>
Date: Thu, 14 May 2020 13:42:25 +0800
Subject: [PATCH] update .batch and .shuffle apis (#4620)

---
 PaddleNLP/lexical_analysis/creator.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/PaddleNLP/lexical_analysis/creator.py b/PaddleNLP/lexical_analysis/creator.py
index bf02b57a..bc8e492f 100644
--- a/PaddleNLP/lexical_analysis/creator.py
+++ b/PaddleNLP/lexical_analysis/creator.py
@@ -105,15 +105,15 @@ def create_pyreader(args,
         # create lac pyreader
         if mode == 'train':
             pyreader.set_sample_list_generator(
-                paddle.batch(
-                    paddle.reader.shuffle(
+                fluid.io.batch(
+                    fluid.io.shuffle(
                         reader.file_reader(file_name),
                         buf_size=args.traindata_shuffle_buffer),
                     batch_size=args.batch_size / device_count),
                 places=place)
         else:
             pyreader.set_sample_list_generator(
-                paddle.batch(
+                fluid.io.batch(
                     reader.file_reader(
                         file_name, mode=mode),
                     batch_size=args.batch_size / device_count),
-- 
GitLab