Bugfix: checkout input_type failed

refactor_docs
Yancey1989 8 years ago
parent 81fb41f097
commit 85502783aa

@ -270,7 +270,7 @@ class CheckWrapper(object):
assert isinstance(each, collections.Sequence) assert isinstance(each, collections.Sequence)
for d in each: for d in each:
assert isinstance(d, float) assert isinstance(d, float)
assert len(each, input_type.dim) assert len(each) == input_type.dim
elif input_type.type == DataType.Index: elif input_type.type == DataType.Index:
assert isinstance(each, int) assert isinstance(each, int)
assert each < input_type.dim assert each < input_type.dim
@ -304,7 +304,7 @@ class CheckInputTypeWrapper(object):
def __call__(self, obj, filename): def __call__(self, obj, filename):
for items in self.generator(obj, filename): for items in self.generator(obj, filename):
try: try:
# dict type is required for input_types when item is dict type # dict type is required for input_types when item is dict type
assert (isinstance(items, dict) and \ assert (isinstance(items, dict) and \
not isinstance(self.input_types, dict))==False not isinstance(self.input_types, dict))==False
yield items yield items

Loading…
Cancel
Save