Commit 98898409 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Small change so that this script work with bob.pad.base databases too

parent 981c5a06
Pipeline #17624 passed with stage
in 24 minutes and 28 seconds
......@@ -61,7 +61,11 @@ def main(command_line_parameters=None):
database.replace_directories(args.database_directories_file)
# get all files
files = database.all_files()
if bob.bio.base.utils.is_argument_available('flat', database.all_files):
files = database.all_files(flat=True)
else:
files = database.all_files()
# filter file ids; convert them to str first
if args.file_ids is not None:
files = [f for f in files if str(f.id) in args.file_ids]
......
Supports Markdown
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