pyrox.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.18.0";
19 format = "setuptools";
20
21 src = fetchPypi {
22 inherit pname version;
23 hash = "sha256-OepK4wxcGCNVoYJEZwExWCFERODOCxjzMzi9gn1PsA8=";
24 };
25
26 propagatedBuildInputs = [
27 numpy
28 scipy
29 protobuf
30 onnx
31 scikit-learn
32 onnxconverter-common
33 ];
34
35 pythonRelaxDeps = [ "scikit-learn" ];
36
37 nativeCheckInputs = [
38 onnxruntime
39 pandas
40 unittestCheckHook
41 ];
42
43 unittestFlagsArray = [
44 "-s"
45 "tests"
46 ];
47
48 # Core dump
49 doCheck = false;
50
51 meta = {
52 description = "Convert scikit-learn models to ONNX";
53 license = with lib.licenses; [ asl20 ];
54 };
55}