COSMIT use zip instead of next in generator expression (scikit-learn#…

…9281)
massich · Jul 13, 2017 · ce6b00a · ce6b00a
1 parent d50a915
commit ce6b00a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/sklearn/multiclass.py b/sklearn/multiclass.py
@@ -267,9 +267,8 @@ def partial_fit(self, X, y, classes=None):
         columns = (col.toarray().ravel() for col in Y.T)
 
         self.estimators_ = Parallel(n_jobs=self.n_jobs)(
-            delayed(_partial_fit_binary)(self.estimators_[i], X,
-                                         next(columns))
-            for i in range(self.n_classes_))
+            delayed(_partial_fit_binary)(estimator, X, column)
+            for estimator, column in izip(self.estimators_, columns))
 
         return self