1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
{
stdenv,
lib,
buildPythonPackage,
fetchPypi,
# build-system
cython,
gfortran,
meson-python,
numpy,
scipy,
# native dependencies
glibcLocales,
llvmPackages,
pytestCheckHook,
pytest-xdist,
pillow,
joblib,
threadpoolctl,
}:
buildPythonPackage rec {
__structuredAttrs = true;
pname = "scikit-learn";
version = "1.7.2";
pyproject = true;
src = fetchPypi {
pname = "scikit_learn";
inherit version;
hash = "sha256-IOnkns0TBZjxyjih2FCQ4aYAFHucAvpvFdactT2Wj9o=";
};
postPatch = ''
substituteInPlace pyproject.toml \
--replace-fail "Cython>=3.0.10,<3.2.0" "Cython>=3.0.10"
substituteInPlace meson.build --replace-fail \
"run_command('sklearn/_build_utils/version.py', check: true).stdout().strip()," \
"'${version}',"
'';
buildInputs = [
numpy.blas
pillow
glibcLocales
]
++ lib.optionals stdenv.cc.isClang [ llvmPackages.openmp ];
nativeBuildInputs = [
gfortran
];
build-system = [
cython
meson-python
numpy
scipy
];
dependencies = [
joblib
numpy
scipy
threadpoolctl
];
pythonRelaxDeps = [
"numpy"
"scipy"
];
nativeCheckInputs = [
pytestCheckHook
pytest-xdist
];
env.LC_ALL = "en_US.UTF-8";
# PermissionError: [Errno 1] Operation not permitted: '/nix/nix-installer'
doCheck = !stdenv.hostPlatform.isDarwin;
disabledTests = [
# Skip test_feature_importance_regression - does web fetch
"test_feature_importance_regression"
# Fail due to new deprecation warnings in scipy
# FIXME: reenable when fixed upstream
"test_logistic_regression_path_convergence_fail"
"test_linalg_warning_with_newton_solver"
"test_newton_cholesky_fallback_to_lbfgs"
# NuSVC memmap tests causes segmentation faults in certain environments
# (e.g. Hydra Darwin machines) related to a long-standing joblib issue
# (https://github.com/joblib/joblib/issues/563). See also:
# https://github.com/scikit-learn/scikit-learn/issues/17582
"NuSVC and memmap"
]
++ lib.optionals stdenv.hostPlatform.isAarch64 [
# doesn't seem to produce correct results?
# possibly relevant: https://github.com/scikit-learn/scikit-learn/issues/25838#issuecomment-2308650816
"test_sparse_input"
];
pytestFlags = [
# verbose build outputs needed to debug hard-to-reproduce hydra failures
"-v"
"--pyargs"
"sklearn"
];
preCheck = ''
cd $TMPDIR
export HOME=$TMPDIR
export OMP_NUM_THREADS=1
'';
pythonImportsCheck = [ "sklearn" ];
meta = {
description = "Set of python modules for machine learning and data mining";
changelog =
let
major = lib.versions.major version;
minor = lib.versions.minor version;
dashVer = lib.replaceStrings [ "." ] [ "-" ] version;
in
"https://scikit-learn.org/stable/whats_new/v${major}.${minor}.html#version-${dashVer}";
homepage = "https://scikit-learn.org";
license = lib.licenses.bsd3;
maintainers = with lib.maintainers; [ davhau ];
};
}
|
