Skip to content

Commit 4e5f636

Browse files
authored
MNT Improve _check_array_api_dispatch docstring (scikit-learn#31831)
1 parent 5c4adff commit 4e5f636

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

sklearn/utils/_array_api.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,10 +124,10 @@ def _get_namespace_device_dtype_ids(param):
124124

125125

126126
def _check_array_api_dispatch(array_api_dispatch):
127-
"""Check that array_api_compat is installed and NumPy version is compatible.
127+
"""Checks that array API support is functional.
128128
129-
array_api_compat follows NEP29, which has a higher minimum NumPy version than
130-
scikit-learn.
129+
In particular scipy needs to be recent enough and the environment variable
130+
needs to be set: SCIPY_ARRAY_API=1.
131131
"""
132132
if not array_api_dispatch:
133133
return

0 commit comments

Comments
 (0)