nixpkgs/pkgs/development/python-modules/speechbrain/default.nix
2023-08-21 09:05:14 +02:00

56 lines
1.0 KiB
Nix

{ buildPythonPackage
, fetchFromGitHub
, huggingface-hub
, hyperpyyaml
, joblib
, lib
, numpy
, packaging
, pythonOlder
, sentencepiece
, scipy
, torch
, torchaudio
, tqdm
}:
buildPythonPackage rec {
pname = "speechbrain";
version = "0.5.15";
format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchFromGitHub {
owner = "speechbrain";
repo = "speechbrain";
rev = "refs/tags/v${version}";
hash = "sha256-d0+3bry69ML65JR8XDppG8RO200ZTTHyd7PrTP7SJkk=";
};
propagatedBuildInputs = [
huggingface-hub
hyperpyyaml
joblib
numpy
packaging
sentencepiece
scipy
torch
torchaudio
tqdm
];
doCheck = false; # requires sox backend
pythonImportsCheck = [ "speechbrain" ];
meta = with lib; {
description = "A PyTorch-based Speech Toolkit";
homepage = "https://speechbrain.github.io";
changelog = "https://github.com/speechbrain/speechbrain/releases/tag/v${version}";
license = licenses.asl20;
maintainers = with maintainers; [ GaetanLepage ];
};
}