removed try/catch

prob-ml · Aug 5, 2024 · e24e63b · e24e63b
1 parent 69363aa
commit e24e63b
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/bliss/cached_dataset.py b/bliss/cached_dataset.py
@@ -170,10 +170,7 @@ def __getitem__(self, index):
             self.buffered_file_index = converted_index
             with open(self.file_paths[converted_index], "rb") as f:
                 self.buffered_data = torch.load(f)
-        try:
-            output_data = self.buffered_data[converted_sub_index]
-        except KeyError:
-            output_data = self.buffered_data
+        output_data = self.buffered_data[converted_sub_index]
         return self.transform(output_data)
 
     def get_chunked_indices(self):