Disable data_format='zip'

ain-soph · Dec 30, 2020 · 6b74b1c · 6b74b1c
1 parent ce1ea5d
commit 6b74b1c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/trojanvision/datasets/imagefolder.py b/trojanvision/datasets/imagefolder.py
@@ -34,6 +34,8 @@ def add_argument(cls, group: argparse._ArgumentGroup):
 
     def __init__(self, data_format: str = 'folder', **kwargs):
         self.data_format: str = data_format
+        if data_format == 'zip':
+            raise NotImplementedError('data_format=zip Not supported! See reason at issues/42 on github.')
         super().__init__(**kwargs)
         self.param_list['imagefolder'] = ['data_format', 'url', 'org_folder_name']
         self.class_to_idx = self.get_class_to_idx()