From 95f0a9de0e820ddcf2c6529a3bcb9d12bba13128 Mon Sep 17 00:00:00 2001
From: zzszmyf <45535243+zzszmyf@users.noreply.github.com>
Date: Wed, 27 Feb 2019 20:18:35 +0800
Subject: [PATCH] Update reader.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

resolve #1794
_reader_creator need default value，if not， the code can't work
---
 fluid/PaddleRec/ctr/reader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fluid/PaddleRec/ctr/reader.py b/fluid/PaddleRec/ctr/reader.py
index affd1d0e..6c58d128 100644
--- a/fluid/PaddleRec/ctr/reader.py
+++ b/fluid/PaddleRec/ctr/reader.py
@@ -13,7 +13,7 @@ class CriteoDataset(Dataset):
         self.continuous_range_ = range(1, 14)
         self.categorical_range_ = range(14, 40)
 
-    def _reader_creator(self, file_list, is_train, trainer_num, trainer_id):
+    def _reader_creator(self, file_list, is_train, trainer_num, trainer_id=0):
         def reader():
             for file in file_list:
                 with open(file, 'r') as f:
-- 
GitLab