summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-07 22:31:55 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-07 22:31:55 +0100
commit36eb7e4de788b8cedbe43029f3c47c3899075dfe (patch)
treebae2153fc7d8894498377b884f4a0c58cb296087 /cli/app/commands/biggan
parent0ba235aca05004abd1f126cbab56dc9196629f15 (diff)
getting latent search updates working...
Diffstat (limited to 'cli/app/commands/biggan')
-rw-r--r--cli/app/commands/biggan/search_class.py2
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