pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, scipy
5, numpy
6, numba
7, scikit-learn
8, pytest
9, pythonOlder
10}:
11
12buildPythonPackage rec {
13 pname = "pygbm";
14 version = "0.1.0";
15 disabled = pythonOlder "3.6";
16
17 src = fetchFromGitHub {
18 owner = "ogrisel";
19 repo = pname;
20 rev = "v${version}";
21 sha256 = "1qg2md86d0z5aa6jn8kj3rxsippsqsccx1dbraspdsdkycncvww3";
22 };
23
24 propagatedBuildInputs = [
25 scipy
26 numpy
27 numba
28 scikit-learn
29 ];
30
31 checkInputs = [
32 pytest
33 ];
34
35 checkPhase = ''
36 # numerical rounding error in test
37 pytest -k "not test_derivatives"
38 '';
39
40 meta = with lib; {
41 description = "Experimental Gradient Boosting Machines in Python";
42 homepage = "https://github.com/ogrisel/pygbm";
43 license = licenses.mit;
44 maintainers = [ maintainers.costrouc ];
45 broken = true;
46 };
47}