Update sklearn/utils/validation.py

alexshacked · glemaitre · web-flow · commit abc3257f79bb · 2020-08-24T14:38:34.000+03:00
Co-authored-by: Guillaume Lemaitre &lt;g.lemaitre58@gmail.com&gt;
diff --git a/sklearn/utils/validation.py b/sklearn/utils/validation.py
@@ -569,8 +569,9 @@ def check_array(array, accept_sparse=False, *, accept_large_sparse=True,
         # DataFrame.sparse only supports `to_coo`
         array = array.sparse.to_coo()
         if array.dtype == np.dtype('object'):
-            unique_dtypes = set([dt.subtype.name for dt in
-                                 array_orig.dtypes])
+            unique_dtypes = set(
+                [dt.subtype.name for dt in array_orig.dtypes]
+            )
             if len(unique_dtypes) > 1:
                 raise ValueError(
                     "Pandas DataFrame with mixed sparse extension arrays "
