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

   1 { lib
   2 , buildPythonPackage
   3 , catboost
   4 , python
   5 , graphviz
   6 , matplotlib
   7 , numpy
   8 , pandas
   9 , plotly
  10 , scipy
  11 , setuptools
  12 , six
  13 , wheel
  14 }:
  15
  16 buildPythonPackage rec {
  17   inherit (catboost) pname version src meta;
  18   format = "pyproject";
  19
  20   sourceRoot = "${src.name}/catboost/python-package";
  21
  22   nativeBuildInputs = [
  23     setuptools
  24     wheel
  25   ];
  26
  27   propagatedBuildInputs = [
  28     graphviz
  29     matplotlib
  30     numpy
  31     pandas
  32     plotly
  33     scipy
  34     six
  35   ];
  36
  37   buildPhase = ''
  38     runHook preBuild
  39
  40     # these arguments must set after bdist_wheel
  41     ${python.pythonOnBuildForHost.interpreter} setup.py bdist_wheel --no-widget --prebuilt-extensions-build-root-dir=${lib.getDev catboost}
  42
  43     runHook postBuild
  44   '';
  45
  46   # setup a test is difficult
  47   doCheck = false;
  48
  49   pythonImportsCheck = [ "catboost" ];
  50 }