Update dataset_tools.py

AntreasAntoniou · Jul 24, 2019 · 6c0490e · 6c0490e
1 parent 0cbfe11
commit 6c0490e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/dataset_tools.py b/utils/dataset_tools.py
@@ -38,7 +38,7 @@ def maybe_unzip_dataset(args):
                 total_files == 3 and 'mini_imagenet_pkl' in datasets[dataset_idx]):
             print("file count is correct")
             done = True
-        elif datasets[dataset_idx] != 'omniglot_dataset' and datasets[dataset_idx] != 'mini_imagenet' and datasets[dataset_idx] and datasets[dataset_idx] != 'mini_imagenet_pkl':
+        elif datasets[dataset_idx] != 'omniglot_dataset' and datasets[dataset_idx] != 'mini_imagenet' and datasets[dataset_idx] != 'mini_imagenet_pkl':
             done = True
             print("using new dataset")