Update simplified_image_classifier.py (#74)

Replace assert by IndexError. This can help to use this with __getitem__ operator in a subscriptable class
paris-saclay-cds · Oct 24, 2017 · dc6ec6c · dc6ec6c
1 parent cea82d3
commit dc6ec6c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rampwf/workflows/simplified_image_classifier.py b/rampwf/workflows/simplified_image_classifier.py
@@ -160,7 +160,8 @@ def load(self, index):
         """
         from skimage.io import imread
 
-        assert 0 <= index < self.nb_examples
+        if index < 0 or index >= self.nb_examples:
+            raise IndexError("list index out of range")
 
         x = self.X_array[index]
         filename = os.path.join(self.folder, '{}'.format(x))