tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 python,
6
7 setuptools,
8 wheel,
9 cython,
10
11 numpy,
12 scipy,
13 scikit-learn,
14}:
15
16buildPythonPackage rec {
17 pname = "quantile-forest";
18 version = "1.4.1";
19 pyproject = true;
20
21 src = fetchFromGitHub {
22 owner = "zillow";
23 repo = "quantile-forest";
24 tag = "v${version}";
25 hash = "sha256-KNHp6/TUy7Aof3P4TNGpsRlNVePrHEc4HFvMqyr4gPg=";
26 };
27
28 build-system = [
29 setuptools
30 cython
31 wheel
32 numpy
33 scipy
34 scikit-learn
35 ];
36
37 dependencies = [
38 numpy
39 scipy
40 scikit-learn
41 ];
42
43 postInstall = ''
44 rm -rf $out/${python.sitePackages}/examples
45 '';
46
47 # need network connection
48 doCheck = false;
49
50 pythonImportsCheck = [ "quantile_forest" ];
51
52 meta = {
53 description = "Quantile Regression Forests compatible with scikit-learn";
54 homepage = "https://github.com/zillow/quantile-forest";
55 changelog = "https://github.com/zillow/quantile-forest/releases/tag/v${version}";
56 license = lib.licenses.asl20;
57 maintainers = with lib.maintainers; [ vizid ];
58 };
59}