58 lines
900 B
Nix
58 lines
900 B
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
numpy,
|
|
scipy,
|
|
protobuf,
|
|
onnx,
|
|
scikit-learn,
|
|
onnxconverter-common,
|
|
onnxruntime,
|
|
pandas,
|
|
unittestCheckHook,
|
|
pythonRelaxDepsHook,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "skl2onnx";
|
|
version = "1.16.0";
|
|
format = "setuptools";
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
hash = "sha256-M3Cz1AZc4txZM4eMMnP0rqQflFzGUUVDsTrS1X82nOU=";
|
|
};
|
|
|
|
propagatedBuildInputs = [
|
|
numpy
|
|
scipy
|
|
protobuf
|
|
onnx
|
|
scikit-learn
|
|
onnxconverter-common
|
|
];
|
|
|
|
nativeBuildInputs = [ pythonRelaxDepsHook ];
|
|
|
|
pythonRelaxDeps = [ "scikit-learn" ];
|
|
|
|
nativeCheckInputs = [
|
|
onnxruntime
|
|
pandas
|
|
unittestCheckHook
|
|
];
|
|
|
|
unittestFlagsArray = [
|
|
"-s"
|
|
"tests"
|
|
];
|
|
|
|
# Core dump
|
|
doCheck = false;
|
|
|
|
meta = {
|
|
description = "Convert scikit-learn models to ONNX";
|
|
license = with lib.licenses; [ asl20 ];
|
|
};
|
|
}
|