From 22a043d900d72dfd9ce7009527bf9f261521efa3 Mon Sep 17 00:00:00 2001 From: tungen Date: Thu, 31 Dec 2015 16:58:47 +0800 Subject: [PATCH] error handling when the input sample is not found in the dataset --- libact/labelers/ideal_labeler.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libact/labelers/ideal_labeler.py b/libact/labelers/ideal_labeler.py index dc89efc..45004a4 100644 --- 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])]