CLN Address comments

thomasjpfan · thomasjpfan · commit 9b9d53439b27 · 2022-03-08T13:52:46.000-05:00
diff --git a/sklearn/ensemble/_voting.py b/sklearn/ensemble/_voting.py
@@ -16,7 +16,6 @@
 from abc import abstractmethod
 
 import numbers
-from itertools import chain
 import numpy as np
 
 from joblib import Parallel
@@ -475,13 +474,10 @@ def get_feature_names_out(self, input_features=None):
 
         # voting == "soft"
         n_classes = len(self.classes_)
-        names_out = chain.from_iterable(
-            [
-                [f"{class_name}_{name}{i}" for i in range(n_classes)]
-                for name in active_names
-            ]
-        )
-        return np.asarray(list(names_out), dtype=object)
+        names_out = [
+            f"{class_name}_{name}{i}" for name in active_names for i in range(n_classes)
+        ]
+        return np.asarray(names_out, dtype=object)
 
 
 class VotingRegressor(RegressorMixin, _BaseVoting):
diff --git a/sklearn/ensemble/tests/test_voting.py b/sklearn/ensemble/tests/test_voting.py
@@ -629,7 +629,7 @@ def test_voting_verbose(estimator, capsys):
 
 
 def test_get_features_names_out_regressor():
-    """Check that get_feature_names_out output for regressor."""
+    """Check get_feature_names_out output for regressor."""
 
     X = [[1, 2], [3, 4], [5, 6]]
     y = [0, 1, 2]
@@ -666,7 +666,7 @@ def test_get_features_names_out_regressor():
     ],
 )
 def test_get_features_names_out_classifier(kwargs, expected_names):
-    """Check get_feature_names_out for different settings."""
+    """Check get_feature_names_out for classifier for different settings."""
     X = [[1, 2], [3, 4], [5, 6], [1, 1.2]]
     y = [0, 1, 2, 0]
 
