修改测试文件

tensnsorflow/record.py

@@ -6,6 +6,7 @@ from __future__ import print_function
 import pandas as pd
 import os
 import glob
+from hdfs import *
 
 import tensorflow as tf
 import numpy as np
@@ -52,6 +53,12 @@ def gen_tfrecords(in_file):
 
 
 def main(_):
+    client = Client("http://nvwa01:50070")
+    path = "/recommend/va"
+    for root, dir, files in client.walk(path):
+        for file in files:
+            print(file)
+
     if not os.path.exists(FLAGS.output_dir):
         os.mkdir(FLAGS.output_dir)
     file_list = glob.glob(os.path.join(FLAGS.input_dir, "*.avro"))
@@ -68,9 +75,4 @@ def main(_):
 if __name__ == "__main__":
     tf.logging.set_verbosity(tf.logging.INFO)
     tf.app.run()
-    from hdfs import *
-    client = Client("http://nvwa01:50070")
-    path = "/recommend/va"
-    for root, dir, files in client.walk(path):
-        for file in files:
-             print(file)
+