Follow-up fix for check_classifiers_train on prescott OpenBLAS (#21704)

ogrisel · web-flow · commit 432ae47beeaf · 2021-11-18T20:08:18.000+01:00
diff --git a/sklearn/utils/estimator_checks.py b/sklearn/utils/estimator_checks.py
@@ -2090,10 +2090,10 @@ def check_classifiers_train(
             # Prudently assume Prescott might be the architecture if it is unknown.
             and info.get("architecture", "prescott").lower() == "prescott"
         )
-
-        X_m, y_m, X_b, y_b = create_memmap_backed_data(
-            data=[X_m, y_m, X_b, y_b], aligned=has_prescott_openblas
-        )
+        X_m = create_memmap_backed_data(data=X_m, aligned=has_prescott_openblas)
+        y_m = create_memmap_backed_data(data=y_m, aligned=has_prescott_openblas)
+        X_b = create_memmap_backed_data(data=X_b, aligned=has_prescott_openblas)
+        y_b = create_memmap_backed_data(data=y_b, aligned=has_prescott_openblas)
 
     problems = [(X_b, y_b)]
     tags = _safe_tags(classifier_orig)
