@@ -202,24 +202,26 @@ def loop_check(callback, item):
202202 for each in item :
203203 callback (each )
204204
205+
205206class CheckInputTypeWrapper (object ):
206207 def __init__ (self , generator , input_types , logger ):
207208 self .generator = generator
208209 self .input_types = input_types
209210 self .logger = logger
210211
211- def __call__ (self , obj , filename ):
212- for items in self .generator (obj , filename ):
213- try :
214- # dict type is required for input_types when item is dict type
215- assert (isinstance (items , dict ) and \
216- not isinstance (self .input_types , dict ))== False
217- yield items
218- except AssertionError as e :
219- self .logger .error (
212+ def __call__ (self , obj , filename ):
213+ for items in self .generator (obj , filename ):
214+ try :
215+ # dict type is required for input_types when item is dict type
216+ assert (isinstance (items , dict ) and \
217+ not isinstance (self .input_types , dict ))== False
218+ yield items
219+ except AssertionError as e :
220+ self .logger .error (
220221 "%s type is required for input type but got %s" %
221222 (repr (type (items )), repr (type (self .input_types ))))
222- raise
223+ raise
224+
223225
224226def provider (input_types = None ,
225227 should_shuffle = None ,
@@ -374,8 +376,8 @@ def __init__(self, file_list, **kwargs):
374376 self .generator = InputOrderWrapper (self .generator ,
375377 self .input_order )
376378 else :
377- self .generator = CheckInputTypeWrapper (self . generator , self . slots ,
378- self .logger )
379+ self .generator = CheckInputTypeWrapper (
380+ self . generator , self . slots , self .logger )
379381 if self .check :
380382 self .generator = CheckWrapper (self .generator , self .slots ,
381383 check_fail_continue ,
0 commit comments