From 998848a22eed3b80133cf7907c73da70e83512ea Mon Sep 17 00:00:00 2001 From: Nick James Date: Mon, 16 Nov 2015 13:23:36 -0800 Subject: [PATCH] fix for #5725 (makes sure numpy array is not compared to string, which will fail in future versions of numpy) --- sklearn/linear_model/randomized_l1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sklearn/linear_model/randomized_l1.py b/sklearn/linear_model/randomized_l1.py index bef06f59911bc..399e5aca83b4d 100644 --- 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,