pkgs/development/python-modules/speechbrain/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5
   6   # dependencies
   7   huggingface-hub,
   8   hyperpyyaml,
   9   joblib,
  10   numpy,
  11   packaging,
  12   sentencepiece,
  13   scipy,
  14   torch,
  15   torchaudio,
  16   tqdm,
  17 }:
  18
  19 buildPythonPackage rec {
  20   pname = "speechbrain";
  21   version = "1.0.2";
  22   pyproject = true;
  23
  24   src = fetchFromGitHub {
  25     owner = "speechbrain";
  26     repo = "speechbrain";
  27     rev = "refs/tags/v${version}";
  28     hash = "sha256-Un7RPxMq1sD7uD3jcw3Bjp+Oo8ld+XC5g2I89gF6jxs=";
  29   };
  30
  31   dependencies = [
  32     huggingface-hub
  33     hyperpyyaml
  34     joblib
  35     numpy
  36     packaging
  37     sentencepiece
  38     scipy
  39     torch
  40     torchaudio
  41     tqdm
  42   ];
  43
  44   doCheck = false; # requires sox backend
  45
  46   pythonImportsCheck = [ "speechbrain" ];
  47
  48   meta = {
  49     description = "PyTorch-based Speech Toolkit";
  50     homepage = "https://speechbrain.github.io";
  51     changelog = "https://github.com/speechbrain/speechbrain/releases/tag/v${version}";
  52     license = lib.licenses.asl20;
  53     maintainers = with lib.maintainers; [ GaetanLepage ];
  54   };
  55 }