Skip to content

F
ffm-baseline
Commit a145a064 authored by 张彦钊's avatar 张彦钊
Browse files
Options

修改测试文件

parent 582bdbf2
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensnsorflow/record.py
View file @ a145a064
...@@ -57,7 +57,7 @@ def main(_): ...@@ -57,7 +57,7 @@ def main(_):
file_list = [] file_list = []
for root, dir, files in client.walk(FLAGS.input_dir): for root, dir, files in client.walk(FLAGS.input_dir):
for file in files: for file in files:
if file[-5:] == "*.avro": if file[-5:] == ".avro":
file_list.append(FLAGS.input_dir+file) file_list.append(FLAGS.input_dir+file)
if not os.path.exists(FLAGS.output_dir): if not os.path.exists(FLAGS.output_dir):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment