change train.py

eda/esmm/Model_pipline/train.py

@@ -421,6 +421,7 @@ if __name__ == "__main__":
         main(tr_files)
     elif FLAGS.task_type == 'infer':
         te_files = ["%s/part-r-00000" % FLAGS.hdfs_dir]
+        print(te_files[0])
         queue_name = te_files[0].split('_')[-1] + "_queue"
         print(queue_name + " task")
         result = main(te_files)