fix for #5725 (makes sure numpy array is not compared to string, which will fail in future versions of numpy)

steph-query · steph-query · commit 998848a22eed · 2015-11-16T13:23:36.000-08:00
diff --git a/sklearn/linear_model/randomized_l1.py b/sklearn/linear_model/randomized_l1.py
@@ -330,7 +330,7 @@ def __init__(self, alpha='aic', scaling=.5, sample_fraction=.75,
     def _make_estimator_and_params(self, X, y):
         assert self.precompute in (True, False, None, 'auto')
         alpha = self.alpha
-        if alpha in ('aic', 'bic'):
+        if isinstance(alpha, six.string_types) and alpha in ('aic', 'bic'):
             model = LassoLarsIC(precompute=self.precompute,
                                 criterion=self.alpha,
                                 max_iter=self.max_iter,
