修改测试文件

tensnsorflow/record.py

@@ -55,9 +55,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)
+    path, dirs, files = client.walk(path)
+    print(files)
+    print(type(files))
+    # 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)