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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , numpy
   5 , onnx
   6 , skl2onnx
   7 # native check inputs
   8 , pytestCheckHook
   9 , pandas
  10 , xgboost
  11 , onnxruntime
  12 , scikit-learn
  13 , pyspark
  14 , lightgbm
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "onnxmltools";
  19   version = "1.12.0";
  20   format = "setuptools";
  21
  22   src = fetchFromGitHub {
  23     owner = "onnx";
  24     repo = "onnxmltools";
  25     rev = "refs/tags/${version}";
  26     hash = "sha256-/UKGo56riLnATcn7kA++QoFkkILVGYBwqRZZ+PYB1/0=";
  27   };
  28
  29   propagatedBuildInputs = [
  30     numpy
  31     onnx
  32     skl2onnx
  33   ];
  34
  35   pythonImportsCheck = [ "onnxmltools" ];
  36
  37   # there are still some dependencies that need to be packaged for the tests to run
  38   doCheck = false;
  39
  40   nativeCheckInputs = [
  41     pytestCheckHook
  42     pandas
  43     xgboost
  44     onnxruntime
  45     scikit-learn
  46     pyspark
  47     lightgbm
  48     # coremltools
  49     # libsvm
  50     # h20
  51   ];
  52
  53   meta = with lib; {
  54     description = "ONNXMLTools enables conversion of models to ONNX";
  55     homepage = "https://github.com/onnx/onnxmltools";
  56     license = licenses.asl20;
  57     maintainers = with maintainers; [ happysalada ];
  58   };
  59 }