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