nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-28 18:03:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.skl2onnx: init at 1.13

Browse Source
This commit is contained in:
Frederik Rietdijk 2022-09-27 16:05:13 +02:00 committed by Frederik Rietdijk
parent 18f7d4c992
commit d5f27c97b8
2 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
pkgs/development/python-modules/skl2onnx/default.nix Normal file
View File

@ -0,0 +1,49 @@
{ lib
, buildPythonPackage
, fetchPypi
, numpy
, scipy
, protobuf
, onnx
, scikit-learn
, onnxconverter-common
, onnxruntime
, pandas
, unittestCheckHook
}:
buildPythonPackage rec {
pname = "skl2onnx";
version = "1.13";
src = fetchPypi {
inherit pname version;
hash = "sha256-XzUva5uFX/rGMFpwfwLH1Db0Nok47pBJCSqVo1ZcJz0=";
};
propagatedBuildInputs = [
numpy
scipy
protobuf
onnx
scikit-learn
onnxconverter-common
];
checkInputs = [
onnxruntime
pandas
unittestCheckHook
];
unittestFlagsArray = [ "-s" "tests" ];
# Core dump
doCheck = false;
meta = {
description = "Convert scikit-learn models to ONNX";
maintainers = with lib.maintainers; [ fridh ];
license = with lib.licenses; [ asl20 ];
};
}

2
pkgs/top-level/python-packages.nix
View File

@ -10167,6 +10167,8 @@ in {
skidl = callPackage ../development/python-modules/skidl { };
skl2onnx = callPackage ../development/python-modules/skl2onnx { };
sklearn-deap = callPackage ../development/python-modules/sklearn-deap { };
skodaconnect = callPackage ../development/python-modules/skodaconnect { };