python3Packages.skl2onnx: init at 1.13

This commit is contained in:
Frederik Rietdijk 2022-09-27 16:05:13 +02:00 committed by Frederik Rietdijk
parent 18f7d4c992
commit d5f27c97b8
2 changed files with 51 additions and 0 deletions

View file

@ -0,0 +1,49 @@
{ lib
, buildPythonPackage
, fetchPypi
, numpy
, scipy
, protobuf
, onnx
, scikit-learn
, onnxconverter-common
, onnxruntime
, pandas
, unittestCheckHook
}:
buildPythonPackage rec {
pname = "skl2onnx";
version = "1.13";
src = fetchPypi {
inherit pname version;
hash = "sha256-XzUva5uFX/rGMFpwfwLH1Db0Nok47pBJCSqVo1ZcJz0=";
};
propagatedBuildInputs = [
numpy
scipy
protobuf
onnx
scikit-learn
onnxconverter-common
];
checkInputs = [
onnxruntime
pandas
unittestCheckHook
];
unittestFlagsArray = [ "-s" "tests" ];
# Core dump
doCheck = false;
meta = {
description = "Convert scikit-learn models to ONNX";
maintainers = with lib.maintainers; [ fridh ];
license = with lib.licenses; [ asl20 ];
};
}

View file

@ -10167,6 +10167,8 @@ in {
skidl = callPackage ../development/python-modules/skidl { };
skl2onnx = callPackage ../development/python-modules/skl2onnx { };
sklearn-deap = callPackage ../development/python-modules/sklearn-deap { };
skodaconnect = callPackage ../development/python-modules/skodaconnect { };