diff --git a/pkgs/development/python-modules/hyperpyyaml/default.nix b/pkgs/development/python-modules/hyperpyyaml/default.nix new file mode 100644 index 000000000000..d66d16dc42f3 --- /dev/null +++ b/pkgs/development/python-modules/hyperpyyaml/default.nix @@ -0,0 +1,39 @@ +{ buildPythonPackage +, fetchFromGitHub +, lib +, pytestCheckHook +, pyyaml +, ruamel-yaml +}: + +buildPythonPackage rec { + pname = "hyperpyyaml"; + version = "1.2.1"; + format = "setuptools"; + + src = fetchFromGitHub { + owner = "speechbrain"; + repo = "hyperpyyaml"; + rev = "refs/tags/v${version}"; + hash = "sha256-tC4kLJAY9MVgjWwU2Qu0rPCVDw7CjKVIciRZgYhnR9I="; + }; + + propagatedBuildInputs = [ + pyyaml + ruamel-yaml + ]; + + nativeCheckInputs = [ + pytestCheckHook + ]; + + pythonImportsCheck = [ "hyperpyyaml" ]; + + meta = with lib; { + description = "Extensions to YAML syntax for better python interaction"; + homepage = "https://github.com/speechbrain/HyperPyYAML"; + changelog = "https://github.com/speechbrain/HyperPyYAML/releases/tag/v${version}"; + license = licenses.asl20; + maintainers = with maintainers; [ GaetanLepage ]; + }; +} diff --git a/pkgs/development/python-modules/speechbrain/default.nix b/pkgs/development/python-modules/speechbrain/default.nix new file mode 100644 index 000000000000..e0a72e331f06 --- /dev/null +++ b/pkgs/development/python-modules/speechbrain/default.nix @@ -0,0 +1,55 @@ +{ buildPythonPackage +, fetchFromGitHub +, huggingface-hub +, hyperpyyaml +, joblib +, lib +, numpy +, packaging +, pythonOlder +, sentencepiece +, scipy +, torch +, torchaudio +, tqdm +}: + +buildPythonPackage rec { + pname = "speechbrain"; + version = "0.5.14"; + format = "setuptools"; + + disabled = pythonOlder "3.7"; + + src = fetchFromGitHub { + owner = "speechbrain"; + repo = "speechbrain"; + rev = "refs/tags/v${version}"; + hash = "sha256-r1q7JO+H7ynfrzlihRTY0PtMGmvwm98BHUZV534ABXw="; + }; + + 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 ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 828108c74c8f..ba3ed72f09f3 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4871,6 +4871,8 @@ self: super: with self; { hyperopt = callPackage ../development/python-modules/hyperopt { }; + hyperpyyaml = callPackage ../development/python-modules/hyperpyyaml { }; + hypothesis-auto = callPackage ../development/python-modules/hypothesis-auto { }; hypothesis = callPackage ../development/python-modules/hypothesis { }; @@ -11723,6 +11725,8 @@ self: super: with self; { spectral-cube = callPackage ../development/python-modules/spectral-cube { }; + speechbrain = callPackage ../development/python-modules/speechbrain { }; + speedtest-cli = callPackage ../development/python-modules/speedtest-cli { }; speg = callPackage ../development/python-modules/speg { };