pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, numpy
5, onnx
6, skl2onnx
7# native check inputs
8, pytestCheckHook
9, pandas
10, xgboost
11, onnxruntime
12, scikit-learn
13, pyspark
14, lightgbm
15}:
16
17buildPythonPackage rec {
18 pname = "onnxmltools";
19 version = "1.11.2";
20 format = "setuptools";
21
22 src = fetchFromGitHub {
23 owner = "onnx";
24 repo = "onnxmltools";
25 rev = "v${version}";
26 hash = "sha256-uLFAGtCDLdMd0SMoonMXFE0kGHuDpwp6IrIbD0t8l4M=";
27 };
28
29 propagatedBuildInputs = [
30 numpy
31 onnx
32 skl2onnx
33 ];
34
35 pythonImportsCheck = [ "onnxmltools" ];
36
37 # there are still some dependencies that need to be packaged for the tests to run
38 doCheck = false;
39
40 nativeCheckInputs = [
41 pytestCheckHook
42 pandas
43 xgboost
44 onnxruntime
45 scikit-learn
46 pyspark
47 lightgbm
48 # coremltools
49 # libsvm
50 # h20
51 ];
52
53 meta = with lib; {
54 description = "ONNXMLTools enables conversion of models to ONNX";
55 homepage = "https://github.com/onnx/onnxmltools";
56 license = licenses.asl20;
57 maintainers = with maintainers; [ happysalada ];
58 };
59}