10000 adding user warning about scipy's dependency · scikit-learn/scikit-learn@21f658c · GitHub
[go: up one dir, main page]
Skip to content

Commit 21f658c

Browse files
EmilyXinyiEmilyXinyi
committed
adding user warning about scipy's dependency
1 parent 100dac3 commit 21f658c

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

‎sklearn/metrics/tests/test_common.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2019,7 +2019,6 @@ def check_array_api_metric_pairwise(metric, array_namespace, device, dtype_name)
20192019
additive_chi2_kernel: [check_array_api_metric_pairwise],
20202020
mean_gamma_deviance: [check_array_api_regression_metric],
20212021
max_error: [check_array_api_regression_metric],
2022-
mean_poisson_deviance: [check_array_api_regression_metric],
20232022
chi2_kernel: [check_array_api_metric_pairwise],
20242023
}
20252024

‎sklearn/utils/_array_api.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
import itertools
44
import math
5+
import os
6+
import warnings
57
from functools import wraps
68

79
import numpy
@@ -103,6 +105,14 @@ def _check_array_api_dispatch(array_api_dispatch):
103105
f"NumPy must be {min_numpy_version} or newer to dispatch array using"
104106
" the API specification"
105107
)
108+
if os.environ.get("SCIPY_ARRAY_API") != "1":
109+
warnings.warn(
110+
(
111+
"Some scikit-learn array API features might rely on enabling "
112+
"SciPy's own support for array API to function properly. "
113+
),
114+
UserWarning,
115+
)
106116

107117

108118
def _single_array_device(array):

0 commit comments

Comments
 (0)
0