diff options
Diffstat (limited to 'cli')
| -rw-r--r-- | cli/app/commands/biggan/search_class.py | 2 | ||||
| -rw-r--r-- | cli/app/utils/cortex_utils.py | 4 |
2 files changed, 3 insertions, 3 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 diff --git a/cli/app/utils/cortex_utils.py b/cli/app/utils/cortex_utils.py index 50e85a4..e4594ff 100644 --- a/cli/app/utils/cortex_utils.py +++ b/cli/app/utils/cortex_utils.py @@ -34,12 +34,12 @@ def find_unprocessed_files(files): datasets = {} unprocessed_files = [] for file in files: - if file['generated'] == 1 and file['type'] == 'image': + if file['generated'] == 1 and file['datatype'] == 'image': fn, ext = os.path.splitext(file['name']) dataset = fn.split('-')[0] datasets[dataset] = file['id'] for file in files: - if file['generated'] == 0 and file['processed'] == 0 and file['type'] == 'image': + if file['generated'] == 0 and file['processed'] == 0 and file['datatype'] == 'image': fn, ext = os.path.splitext(file['name']) dataset = fn.split('-')[0] if dataset not in datasets: |