修改测试文件

a145a064 · 张彦钊 · 582bdbf2 · a145a064
Commit a145a064 authored May 05, 2019 by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

record.py tensnsorflow/record.py +1 -1

No files found.
--- a/tensnsorflow/record.py
+++ b/tensnsorflow/record.py
@@ -57,7 +57,7 @@ def main(_):
    file_list = []
    for root, dir, files in client.walk(FLAGS.input_dir):
        for file in files:
-            if file[-5:] == "*.avro":
+            if file[-5:] == ".avro":
                file_list.append(FLAGS.input_dir+file)

    if not os.path.exists(FLAGS.output_dir):