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

57 lines
899 B
Nix

{ lib
, buildPythonPackage
, fetchPypi
, numpy
, scipy
, protobuf
, onnx
, scikit-learn
, onnxconverter-common
, onnxruntime
, pandas
, unittestCheckHook
, pythonRelaxDepsHook
}:
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
];
nativeBuildInputs = [
pythonRelaxDepsHook
];
pythonRelaxDeps = [ "scikit-learn" ];
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 ];
};
}