diff options
Diffstat (limited to 'pkgs/development/python-modules/skl2onnx/default.nix')
-rw-r--r-- | pkgs/development/python-modules/skl2onnx/default.nix | 37 |
1 files changed, 19 insertions, 18 deletions
diff --git a/pkgs/development/python-modules/skl2onnx/default.nix b/pkgs/development/python-modules/skl2onnx/default.nix index 3cd47324cb731..e696834b13e1d 100644 --- a/pkgs/development/python-modules/skl2onnx/default.nix +++ b/pkgs/development/python-modules/skl2onnx/default.nix @@ -1,16 +1,17 @@ -{ lib -, buildPythonPackage -, fetchPypi -, numpy -, scipy -, protobuf -, onnx -, scikit-learn -, onnxconverter-common -, onnxruntime -, pandas -, unittestCheckHook -, pythonRelaxDepsHook +{ + lib, + buildPythonPackage, + fetchPypi, + numpy, + scipy, + protobuf, + onnx, + scikit-learn, + onnxconverter-common, + onnxruntime, + pandas, + unittestCheckHook, + pythonRelaxDepsHook, }: buildPythonPackage rec { @@ -32,9 +33,7 @@ buildPythonPackage rec { onnxconverter-common ]; - nativeBuildInputs = [ - pythonRelaxDepsHook - ]; + nativeBuildInputs = [ pythonRelaxDepsHook ]; pythonRelaxDeps = [ "scikit-learn" ]; @@ -44,14 +43,16 @@ buildPythonPackage rec { unittestCheckHook ]; - unittestFlagsArray = [ "-s" "tests" ]; + 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 ]; }; } |