modAL.utils.combination.make_query_strategy bug fixed

cosmic-cortex · cosmic-cortex · commit fea5343a94cb · 2018-04-17T14:38:39.000+02:00
diff --git a/modAL/utils/combination.py b/modAL/utils/combination.py
@@ -76,7 +76,7 @@ def make_query_strategy(utility_measure, selector):
 
     def query_strategy(classifier, X):
         utility = utility_measure(classifier, X)
-        query_idx, query_instance = selector(utility)
-        return query_idx, query_instance
+        query_idx = selector(utility)
+        return query_idx, X[query_idx]
 
     return query_strategy
