diff --git a/openfe/openfe.py b/openfe/openfe.py index 8d7596c..4fd4d56 100644 --- a/openfe/openfe.py +++ b/openfe/openfe.py @@ -715,8 +715,10 @@ def _calculate_and_evaluate(self, candidate_features, train_idx, val_idx): length = int(np.ceil(len(candidate_features) / self.n_jobs / 4)) n = int(np.ceil(len(candidate_features) / length)) random.shuffle(candidate_features) - for f in candidate_features: - f.delete() + # for f in candidate_features: + # f.delete() + for f in self.candidate_features: + del f with ProcessPoolExecutor(max_workers=self.n_jobs) as ex: with tqdm(total=n) as progress: for i in range(n):