diff --git a/pkgs/development/python-modules/quantile-forest/default.nix b/pkgs/development/python-modules/quantile-forest/default.nix new file mode 100644 index 000000000000..3b6bd12cd4c8 --- /dev/null +++ b/pkgs/development/python-modules/quantile-forest/default.nix @@ -0,0 +1,62 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + python, + pythonOlder, + + setuptools, + wheel, + cython, + + numpy, + scipy, + scikit-learn, +}: + +buildPythonPackage rec { + pname = "quantile-forest"; + version = "1.3.4"; + pyproject = true; + + disabled = pythonOlder "3.8"; + + src = fetchFromGitHub { + owner = "zillow"; + repo = "quantile-forest"; + rev = "refs/tags/v${version}"; + hash = "sha256-hzLJq0y+qjc48PfHW3i73x9safGOy0V1HEQ5WR8IXpI="; + }; + + build-system = [ + setuptools + cython + wheel + numpy + scipy + scikit-learn + ]; + + dependencies = [ + numpy + scipy + scikit-learn + ]; + + postInstall = '' + rm -rf $out/${python.sitePackages}/examples + ''; + + # need network connection + doCheck = false; + + pythonImportsCheck = [ "quantile_forest" ]; + + meta = with lib; { + description = "Quantile Regression Forests compatible with scikit-learn"; + homepage = "https://github.com/zillow/quantile-forest"; + changelog = "https://github.com/zillow/quantile-forest/releases/tag/v${version}"; + license = licenses.asl20; + maintainers = with maintainers; [ vizid ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 21d881d64262..8c81a1a13d82 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -12865,6 +12865,8 @@ self: super: with self; { quandl = callPackage ../development/python-modules/quandl { }; + quantile-forest = callPackage ../development/python-modules/quantile-forest { }; + quantities = callPackage ../development/python-modules/quantities { }; quantiphy = callPackage ../development/python-modules/quantiphy { };