From e90700695394b89af6f2d14d6b541c337f953348 Mon Sep 17 00:00:00 2001 From: Olivier Grisel Date: Thu, 18 Nov 2021 19:35:30 +0100 Subject: [PATCH] Follow-up fix for check_classifiers_train on prescott OpenBLAS --- sklearn/utils/estimator_checks.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/sklearn/utils/estimator_checks.py b/sklearn/utils/estimator_checks.py index 83deff0d5d7e0..074c9e38b77a2 100644 --- 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)