修改测试文件

ed15482a · 张彦钊 · 7d70885d · ed15482a
Commit ed15482a authored May 05, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

record.py tensnsorflow/record.py +6 -3

No files found.
--- a/tensnsorflow/record.py
+++ b/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):
+    path, dirs, files = client.walk(path)
-        for file in files:
+    print(files)
-            print(file)
+    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)