Follow comments

Author: reyoung

demo/mnist/api_train_v2.py

@@ -113,7 +113,7 @@ def event_handler(event):
     test_creator = paddle.dataset.mnist.test()
     test_data = []
     for item in test_creator():
-        test_data.append(item[0])
+        test_data.append((item[0], ))
         if len(test_data) == 100:
             break
 

python/paddle/v2/inference.py

@@ -43,10 +43,7 @@ def iter_infer(self,
 
             def __reader_impl__():
                 for each_sample in input:
-                    if len(reader_dict) == 1:
-                        yield [each_sample]
-                    else:
-                        yield each_sample
+                    yield each_sample
 
             reader = minibatch.batch(__reader_impl__, batch_size=batch_size)
         else: