pkgs/development/python-modules/pygbm/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , scipy
   5 , numpy
   6 , numba
   7 , scikitlearn
   8 , pytest
   9 , pythonOlder
  10 }:
  11
  12 buildPythonPackage 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     scikitlearn
  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 }