added .csv image loading utils

Lightning-Universe · Feb 14, 2021 · dda2834 · dda2834
1 parent 2b213fb
commit dda2834
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/vision/classification/data.py b/flash/vision/classification/data.py
@@ -245,7 +245,7 @@ def before_collate(self, samples: Any) -> Any:
                     output = self._loader(sample)
                     transform = self._valid_transform if self._use_valid_transform else self._train_transform
                     outputs.append(transform(output))
-                except UnidentifiedImageError as e:
+                except UnidentifiedImageError:
                     print(f'Skipping: could not read file {sample}')
 
             return outputs