From dda28349310862b24ab3fe6aa3b651a4145beb6f Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sun, 14 Feb 2021 00:10:43 -0500
Subject: [PATCH] added .csv image loading utils

---
 flash/vision/classification/data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flash/vision/classification/data.py b/flash/vision/classification/data.py
index 766da2460b..fcbfb5e5a1 100644
--- 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