leave one out bug fix

Author: jhdlee

brainbox/modeling/utils.py

@@ -69,8 +69,8 @@ def __init__(self, model, n_features_to_select=None,
         self.scoring = scoring
         self.delta_scores = pd.DataFrame(index=self.model.clu_ids)
         self.trlabels = self.design.trlabels
-        self.train = np.isin(self.trlabels, self.model.traininds).flatten() if train is None else train
-        self.test = ~self.train if test is None else test
+        self.train = np.isin(self.trlabels, self.model.traininds).flatten() if train is None else np.isin(self.trlabels, train).flatten()
+        self.test = ~self.train if test is None else np.isin(self.trlabels, test).flatten()
         self.features = np.array(list(self.design.covar.keys()))
 
     def fit(self, progress=False):