Follow comments

demo/mnist/api_train_v2.py

@@ -113,7 +113,7 @@ def main():
     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 @@ class Inference(object):
 
             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: