pkgs/development/python-modules/onnxruntime-tools/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   coloredlogs,
   6   numpy,
   7   onnx,
   8   packaging,
   9   psutil,
  10   py-cpuinfo,
  11   py3nvml,
  12   sympy,
  13 }:
  14
  15 buildPythonPackage rec {
  16   pname = "onnxruntime-tools";
  17   version = "1.7.0";
  18   format = "wheel";
  19
  20   # the build distribution doesn't work at all, it seems to expect the same structure
  21   # as the github source repo.
  22   # The github source wasn't immediately obvious how to build for this subpackage.
  23   src = fetchPypi {
  24     pname = "onnxruntime_tools";
  25     inherit version;
  26     format = "wheel";
  27     dist = "py3";
  28     python = "py3";
  29     hash = "sha256-Hf+Ii1xIKsW8Yn8S4QhEX+/LPWAMQ/Y2M5dTFv5hetg=";
  30   };
  31
  32   propagatedBuildInputs = [
  33     coloredlogs
  34     numpy
  35     onnx
  36     packaging
  37     psutil
  38     py-cpuinfo
  39     py3nvml
  40     sympy
  41   ];
  42
  43   pythonImportsCheck = [ "onnxruntime_tools" ];
  44
  45   meta = with lib; {
  46     description = "Transformers Model Optimization Tool of ONNXRuntime";
  47     homepage = "https://pypi.org/project/onnxruntime-tools/";
  48     license = with licenses; [ mit ];
  49     maintainers = with maintainers; [ happysalada ];
  50   };
  51 }