Update features/feature_engineer.py

features/feature_engineer.py

@@ -322,7 +322,14 @@ class AdvancedFeatureEngineer(BaseEstimator, TransformerMixin):
         # Apply feature selection to names if applicable
         if self.feature_selector is not None:
             selected_indices = self.feature_selector.get_support()
-            self.feature_names_ = [name for i, name in enumerate(self.feature_names_) if selected_indices[i]]
+            # Add bounds checking to prevent IndexError
+            if len(selected_indices) == len(self.feature_names_):
+                self.feature_names_ = [name for i, name in enumerate(self.feature_names_) if selected_indices[i]]
+            else:
+                logger.warning(f"Mismatch: {len(selected_indices)} selected_indices vs {len(self.feature_names_)} feature_names")
+                # Use the shorter length to avoid index errors
+                min_length = min(len(selected_indices), len(self.feature_names_))
+                self.feature_names_ = [name for i, name in enumerate(self.feature_names_[:min_length]) if i < len(selected_indices) and selected_indices[i]]
     
     def _calculate_feature_importance(self):
         """Calculate feature importance scores"""