diff options
Diffstat (limited to 'cli/app/commands/biggan/search_class.py')
| -rw-r--r-- | cli/app/commands/biggan/search_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index 4e3973b..ef86613 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -41,7 +41,7 @@ def cli(ctx, opt_folder_id, opt_fp_in, opt_dims, opt_steps, opt_limit, opt_video if opt_folder_id != 0: folder = cortex_folder(opt_folder_id) files = download_cortex_files(opt_folder_id) - unprocessed_files = [file for file in files if file['generated'] == 0 and file['type'] == 'image'] + unprocessed_files = [file for file in files if file['generated'] == 0 and file['datatype'] == 'image'] if len(unprocessed_files) == 0: print("All files processed, nothing to do") return |