提交 32ecc191 编写于 作者: Y Yancey 提交者: GitHub

Merge pull request #2100 from Yancey1989/fix2060

Bugfix: checkout input_type failed
...@@ -270,7 +270,7 @@ class CheckWrapper(object): ...@@ -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): ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册