python3Packages.onnx-asr: init at 0.10.2 (#490362)

pkgs/development/python-modules/onnx-asr/default.nix

@@ -0,0 +1,83 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+
+  # build-system
+  hatchling,
+  hatch-vcs,
+
+  # build-time deps for the custom hatch build hook that generates
+  # ONNX preprocessor models (listed in pyproject.toml [dependency-groups] build)
+  numpy,
+  onnx,
+  onnxscript,
+  torch,
+  torchaudio,
+
+  # dependencies
+  onnxruntime,
+
+  # optional-dependencies
+  huggingface-hub,
+}:
+
+buildPythonPackage (finalAttrs: {
+  pname = "onnx-asr";
+  version = "0.10.2";
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "istupakov";
+    repo = "onnx-asr";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-KumdelY9oNMAEBSGVdvbBH6SYi93n2cA/eEqaE8MmIU=";
+  };
+
+  build-system = [
+    hatchling
+    hatch-vcs
+    # The custom hatch build hook (hatch_build.py) generates ONNX preprocessor
+    # models at build time using these dependencies.
+    numpy
+    onnx
+    onnxscript
+    torch
+    torchaudio
+  ];
+
+  dependencies = [
+    numpy
+    onnxruntime
+  ];
+
+  optional-dependencies = {
+    cpu = [
+      onnxruntime
+    ];
+    hub = [
+      huggingface-hub
+    ];
+    # gpu extra installs onnxruntime-gpu; in nixpkgs users should use
+    # onnxruntime built with cudaSupport instead
+    gpu = [
+      onnxruntime
+    ];
+  };
+
+  # Most tests require downloading models from Hugging Face
+  doCheck = false;
+
+  pythonImportsCheck = [
+    "onnx_asr"
+  ];
+
+  meta = {
+    description = "Lightweight Automatic Speech Recognition using ONNX models";
+    homepage = "https://github.com/istupakov/onnx-asr";
+    changelog = "https://github.com/istupakov/onnx-asr/releases/tag/${finalAttrs.src.tag}";
+    license = lib.licenses.mit;
+    mainProgram = "onnx-asr";
+    maintainers = with lib.maintainers; [ jaredmontoya ];
+  };
+})

pkgs/top-level/python-packages.nix

@@ -11301,6 +11301,8 @@ self: super: with self; {
     };
   };
 
+  onnx-asr = callPackage ../development/python-modules/onnx-asr { };
+
   onnx-ir = callPackage ../development/python-modules/onnx-ir { };
 
   onnxconverter-common = callPackage ../development/python-modules/onnxconverter-common { };