added custom return value

Author: stsouko

chytorch/utils/data/_utils.py

@@ -95,14 +95,15 @@ class SuppressException(Dataset):
     """
     Catch exceptions in wrapped dataset and return None instead
     """
-    def __init__(self, dataset):
+    def __init__(self, dataset, *, default=None):
         self.dataset = dataset
+        self.default = default
 
     def __getitem__(self, item):
         try:
             return self.dataset[item]
         except Exception:
-            pass
+            return self.default
 
     def __len__(self):
         return len(self.dataset)