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

   1 { buildPythonPackage
   2 , fetchFromGitHub
   3 , huggingface-hub
   4 , hyperpyyaml
   5 , joblib
   6 , lib
   7 , numpy
   8 , packaging
   9 , pythonOlder
  10 , sentencepiece
  11 , scipy
  12 , torch
  13 , torchaudio
  14 , tqdm
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "speechbrain";
  19   version = "1.0.0";
  20   format = "setuptools";
  21
  22   disabled = pythonOlder "3.7";
  23
  24   src = fetchFromGitHub {
  25     owner = "speechbrain";
  26     repo = "speechbrain";
  27     rev = "refs/tags/v${version}";
  28     hash = "sha256-s23RHxqr+ZCWPYNchFpyYa01mY+Bum3thgNbJP8rAmA=";
  29   };
  30
  31   propagatedBuildInputs = [
  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 = with lib; {
  49     description = "A PyTorch-based Speech Toolkit";
  50     homepage = "https://speechbrain.github.io";
  51     changelog = "https://github.com/speechbrain/speechbrain/releases/tag/v${version}";
  52     license = licenses.asl20;
  53     maintainers = with maintainers; [ GaetanLepage ];
  54   };
  55 }