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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   numpy,
   6   scipy,
   7   protobuf,
   8   onnx,
   9   scikit-learn,
  10   onnxconverter-common,
  11   onnxruntime,
  12   pandas,
  13   unittestCheckHook,
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "skl2onnx";
  18   version = "1.17.0";
  19   format = "setuptools";
  20
  21   src = fetchPypi {
  22     inherit pname version;
  23     hash = "sha256-cSfchORw9In2gJTM//mlqBW2CfcA1D5wjm9lijOwZAM=";
  24   };
  25
  26   propagatedBuildInputs = [
  27     numpy
  28     scipy
  29     protobuf
  30     onnx
  31     scikit-learn
  32     onnxconverter-common
  33   ];
  34
  35
  36   pythonRelaxDeps = [ "scikit-learn" ];
  37
  38   nativeCheckInputs = [
  39     onnxruntime
  40     pandas
  41     unittestCheckHook
  42   ];
  43
  44   unittestFlagsArray = [
  45     "-s"
  46     "tests"
  47   ];
  48
  49   # Core dump
  50   doCheck = false;
  51
  52   meta = {
  53     description = "Convert scikit-learn models to ONNX";
  54     license = with lib.licenses; [ asl20 ];
  55   };
  56 }