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