bug fix

7c5e1ade · Your Name · 482c43ed · 7c5e1ade
Commit 7c5e1ade authored Jun 06, 2019 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

dist_predict.py eda/esmm/Model_pipline/dist_predict.py +1 -1

No files found.
--- a/eda/esmm/Model_pipline/dist_predict.py
+++ b/eda/esmm/Model_pipline/dist_predict.py
@@ -195,7 +195,7 @@ if __name__ == "__main__":
    print(test.collect())

    tf.logging.set_verbosity(tf.logging.INFO)
-    te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/nearby/part-r-00000"]
+    te_files = [["hdfs://172.16.32.4:8020/strategy/esmm/nearby/part-r-00000"]]
    rdd_te_files = spark.sparkContext.parallelize(te_files)
    now_time = rdd_te_files.map(lambda x: main(x))
    print(now_time)