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

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