646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
57 lines
1.1 KiB
Nix
57 lines
1.1 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
|
|
# dependencies
|
|
coloredlogs,
|
|
numpy,
|
|
onnx,
|
|
packaging,
|
|
psutil,
|
|
py-cpuinfo,
|
|
py3nvml,
|
|
sympy,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "onnxruntime-tools";
|
|
version = "1.7.0";
|
|
format = "wheel";
|
|
|
|
# the build distribution doesn't work at all, it seems to expect the same structure
|
|
# as the github source repo.
|
|
# The github source wasn't immediately obvious how to build for this subpackage.
|
|
src = fetchPypi {
|
|
pname = "onnxruntime_tools";
|
|
inherit version;
|
|
format = "wheel";
|
|
dist = "py3";
|
|
python = "py3";
|
|
hash = "sha256-Hf+Ii1xIKsW8Yn8S4QhEX+/LPWAMQ/Y2M5dTFv5hetg=";
|
|
};
|
|
|
|
dependencies = [
|
|
coloredlogs
|
|
numpy
|
|
onnx
|
|
packaging
|
|
psutil
|
|
py-cpuinfo
|
|
py3nvml
|
|
sympy
|
|
];
|
|
|
|
pythonImportsCheck = [ "onnxruntime_tools" ];
|
|
|
|
# no tests
|
|
doCheck = false;
|
|
|
|
meta = {
|
|
description = "Transformers Model Optimization Tool of ONNXRuntime";
|
|
homepage = "https://pypi.org/project/onnxruntime-tools/";
|
|
license = with lib.licenses; [ mit ];
|
|
maintainers = with lib.maintainers; [ happysalada ];
|
|
};
|
|
}
|