depot/third_party/nixpkgs/pkgs/development/python-modules/skl2onnx/default.nix
Default email c7f94ff3ce Project import generated by Copybara.
GitOrigin-RevId: b85ed9dcbf187b909ef7964774f8847d554fab3b
2023-08-22 22:05:09 +02:00

56 lines
907 B
Nix

{ lib
, buildPythonPackage
, fetchPypi
, numpy
, scipy
, protobuf
, onnx
, scikit-learn
, onnxconverter-common
, onnxruntime
, pandas
, unittestCheckHook
, pythonRelaxDepsHook
}:
buildPythonPackage rec {
pname = "skl2onnx";
version = "1.15.0";
src = fetchPypi {
inherit pname version;
hash = "sha256-BbLCZDrQNX7B6mhNE4Q4ot9lffgo5X0Hy3jC52viDjc=";
};
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";
maintainers = with lib.maintainers; [ fridh ];
license = with lib.licenses; [ asl20 ];
};
}