From 85502783aaa85a43c82a851e9944f535f09c394f Mon Sep 17 00:00:00 2001
From: Yancey1989 <yancey1989@gmail.com>
Date: Thu, 11 May 2017 14:24:33 +0800
Subject: [PATCH] Bugfix: checkout input_type failed
---
python/paddle/trainer/PyDataProvider2.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/paddle/trainer/PyDataProvider2.py b/python/paddle/trainer/PyDataProvider2.py
index 0e752c117c..a36f0ebfdc 100644
--- a/python/paddle/trainer/PyDataProvider2.py
+++ b/python/paddle/trainer/PyDataProvider2.py
@@ -270,7 +270,7 @@ class CheckWrapper(object):
assert isinstance(each, collections.Sequence)
for d in each:
assert isinstance(d, float)
- assert len(each, input_type.dim)
+ assert len(each) == input_type.dim
elif input_type.type == DataType.Index:
assert isinstance(each, int)
assert each < input_type.dim
@@ -304,7 +304,7 @@ class CheckInputTypeWrapper(object):
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
+ # 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