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