change test file

eda/esmm/Model_pipline/feature_engineering.py

@@ -87,7 +87,7 @@ def train_yingshe(y):
                         value_map[x[17]], value_map[x[18]], value_map[x[19]], value_map[x[20]], value_map[x[21]],
                         value_map[x[22]], value_map[x[23]], value_map[x[24]], value_map[x[25]], value_map[x[26]]])
         tmp.append(i)
-    return tmp
+    return iter(tmp)
 
 def pre_yingshe(y):
     tmp = []
@@ -108,7 +108,7 @@ def pre_yingshe(y):
           value_map.get(x[29], 15)
           ])
         tmp.append(i)
-    return tmp
+    return iter(tmp)
 
 def feature_engineer():
     apps_number, app_list_map, level2_number, leve2_map, level3_number, leve3_map = get_map()