diff --git a/pkgs/development/python-modules/onnxslim/default.nix b/pkgs/development/python-modules/onnxslim/default.nix index a2e35dc06d0e..72d0af96ad07 100644 --- a/pkgs/development/python-modules/onnxslim/default.nix +++ b/pkgs/development/python-modules/onnxslim/default.nix @@ -1,9 +1,13 @@ { lib, buildPythonPackage, - fetchPypi, + fetchFromGitHub, + + # build-system setuptools, - wheel, + + # dependencies + colorama, onnx, packaging, sympy, @@ -11,28 +15,31 @@ buildPythonPackage rec { pname = "onnxslim"; - version = "0.1.62"; + version = "0.1.68"; pyproject = true; - src = fetchPypi { - inherit pname version; - hash = "sha256-f9SYFsqIM1h+W62ut3gezrNvv02mMVM/Q9UONJsE2Wg="; + src = fetchFromGitHub { + owner = "inisis"; + repo = "OnnxSlim"; + tag = "v${version}"; + hash = "sha256-1jYdti/Ug1/PoNKN57lxKrHqRTE9BB35+wn8jTLZVpo="; }; build-system = [ setuptools - wheel ]; dependencies = [ + colorama onnx packaging sympy ]; - pythonImportsCheck = [ - "onnxslim" - ]; + pythonImportsCheck = [ "onnxslim" ]; + + # __main__.py: error: the following arguments are required: --model-dir + doCheck = false; meta = { description = "Toolkit to Help Optimize Onnx Model";