error handling when the input sample is not found in the dataset

ntucllab · Dec 31, 2015 · 22a043d · 22a043d
1 parent b99dcf2
commit 22a043d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libact/labelers/ideal_labeler.py b/libact/labelers/ideal_labeler.py
@@ -26,4 +26,6 @@ def __init__(self, dataset, **kwargs):
         self.y = y
 
     def label(self, feature):
-        return self.y[np.where([np.array_equal(x, feature) for x in self.X])[0]]
+        if np.count_nonzero([np.array_equal(x, feature) for x in self.X])==0
+            raise ValueError("No matching sample found in the dataset.")
+        return self.y[np.where([np.array_equal(x, feature) for x in self.X])]