diff --git a/pkgs/development/python-modules/onnxslim/default.nix b/pkgs/development/python-modules/onnxslim/default.nix new file mode 100644 index 000000000000..b115fc4d1c93 --- /dev/null +++ b/pkgs/development/python-modules/onnxslim/default.nix @@ -0,0 +1,43 @@ +{ + lib, + buildPythonPackage, + fetchPypi, + setuptools, + wheel, + onnx, + packaging, + sympy, +}: + +buildPythonPackage rec { + pname = "onnxslim"; + version = "0.1.51"; + pyproject = true; + + src = fetchPypi { + inherit pname version; + hash = "sha256-RH0UnMRJvs+lT/YblJcSE1TflAosJLB3XwD8XhqSdsw="; + }; + + build-system = [ + setuptools + wheel + ]; + + dependencies = [ + onnx + packaging + sympy + ]; + + pythonImportsCheck = [ + "onnxslim" + ]; + + meta = { + description = "A Toolkit to Help Optimize Onnx Model"; + homepage = "https://pypi.org/project/onnxslim/"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ ferrine ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index edb7b4e06ed4..4bbf07e0ae96 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -10319,6 +10319,8 @@ self: super: with self; { onnxruntime-tools = callPackage ../development/python-modules/onnxruntime-tools { }; + onnxslim = callPackage ../development/python-modules/onnxslim { }; + onvif-zeep = callPackage ../development/python-modules/onvif-zeep { }; onvif-zeep-async = callPackage ../development/python-modules/onvif-zeep-async { };