From 454ca01af37292a83f869bd9d7d0d3ae3dd7a4f5 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 13 Dec 2016 15:50:36 +0800
Subject: [PATCH] Fix Travis-CI build error.

* error because merge of #711. The old issue don't check pre-commit
  hooks.
---
 python/paddle/trainer/PyDataProvider2.py | 26 +++++++++++++-----------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/python/paddle/trainer/PyDataProvider2.py b/python/paddle/trainer/PyDataProvider2.py
index f3e3fbf483..6618153df3 100644
--- a/python/paddle/trainer/PyDataProvider2.py
+++ b/python/paddle/trainer/PyDataProvider2.py
@@ -202,24 +202,26 @@ class CheckWrapper(object):
         for each in item:
             callback(each)
 
+
 class CheckInputTypeWrapper(object):
     def __init__(self, generator, input_types, logger):
         self.generator = generator
         self.input_types = input_types
         self.logger = logger
 
-    def __call__(self, obj, filename): 
-       for items in self.generator(obj, filename):
-           try:
-               # dict type is required for input_types when item is dict type 
-               assert (isinstance(items, dict) and \
-                       not isinstance(self.input_types, dict))==False
-               yield items
-           except AssertionError as e:
-               self.logger.error(
+    def __call__(self, obj, filename):
+        for items in self.generator(obj, filename):
+            try:
+                # dict type is required for input_types when item is dict type 
+                assert (isinstance(items, dict) and \
+                        not isinstance(self.input_types, dict))==False
+                yield items
+            except AssertionError as e:
+                self.logger.error(
                     "%s type is required for input type but got %s" %
                     (repr(type(items)), repr(type(self.input_types))))
-               raise
+                raise
+
 
 def provider(input_types=None,
              should_shuffle=None,
@@ -374,8 +376,8 @@ def provider(input_types=None,
                     self.generator = InputOrderWrapper(self.generator,
                                                        self.input_order)
                 else:
-                    self.generator = CheckInputTypeWrapper(self.generator, self.slots,
-                                                           self.logger) 
+                    self.generator = CheckInputTypeWrapper(
+                        self.generator, self.slots, self.logger)
                 if self.check:
                     self.generator = CheckWrapper(self.generator, self.slots,
                                                   check_fail_continue,
-- 
GitLab