test

eda/esmm/Model_pipline/dist_predict.py

@@ -218,7 +218,7 @@ if __name__ == "__main__":
     print(rdd_te_files.collect())
     print("-" * 100)
     indices = rdd_te_files.repartition(100).map(lambda x: main(x))
-    print(indices.take(1))
+    print(indices.collect())
 
 
     b = time.time()