diff --git a/pkgs/development/python-modules/onnxmltools/default.nix b/pkgs/development/python-modules/onnxmltools/default.nix new file mode 100644 index 00000000000..97eaf68254b --- /dev/null +++ b/pkgs/development/python-modules/onnxmltools/default.nix @@ -0,0 +1,59 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, numpy +, onnx +, skl2onnx +# native check inputs +, pytestCheckHook +, pandas +, xgboost +, onnxruntime +, scikit-learn +, pyspark +, lightgbm +}: + +buildPythonPackage rec { + pname = "onnxmltools"; + version = "1.11.2"; + format = "setuptools"; + + src = fetchFromGitHub { + owner = "onnx"; + repo = "onnxmltools"; + rev = "v${version}"; + hash = "sha256-uLFAGtCDLdMd0SMoonMXFE0kGHuDpwp6IrIbD0t8l4M="; + }; + + propagatedBuildInputs = [ + numpy + onnx + skl2onnx + ]; + + pythonImportsCheck = [ "onnxmltools" ]; + + # there are still some dependencies that need to be packaged for the tests to run + doCheck = false; + + nativeCheckInputs = [ + pytestCheckHook + pandas + xgboost + onnxruntime + scikit-learn + pyspark + lightgbm + # coremltools + # libsvm + # h20 + ]; + + meta = with lib; { + description = "ONNXMLTools enables conversion of models to ONNX"; + homepage = "https://github.com/onnx/onnxmltools"; + license = licenses.asl20; + maintainers = with maintainers; [ happysalada ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index c319990a5fe..a34506c4b85 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -7315,6 +7315,8 @@ self: super: with self; { onnxconverter-common = callPackage ../development/python-modules/onnxconverter-common { }; + onnxmltools = callPackage ../development/python-modules/onnxmltools { }; + onnxruntime = callPackage ../development/python-modules/onnxruntime { onnxruntime = pkgs.onnxruntime.override { python3Packages = self;