tts: 0.4.2 -> 0.5.0

pkgs/tools/audio/tts/default.nix

@@ -15,23 +15,32 @@
 
 python3.pkgs.buildPythonApplication rec {
   pname = "tts";
-  version = "0.4.2";
+  version = "0.5.0";
+  format = "setuptools";
 
   src = fetchFromGitHub {
     owner = "coqui-ai";
     repo = "TTS";
     rev = "v${version}";
-    sha256 = "sha256-8a68iFbqqKwtZvufu1Vnv6hGHIQ3HU34wjuQsmr1NUA=";
+    sha256 = "sha256-9fNYNhHS9wqrk2bZnrkkGU1OaDu/16RA8fz+Zj9xsyQ=";
   };
 
-  postPatch = ''
+  postPatch = let
-    sed -i requirements.txt \
+    relaxedConstraints = [
-      -e 's!librosa==[^"]*!librosa!' \
+      "gruut"
-      -e 's!gruut\[.*\]~=2.0.0!gruut!' \
+      "librosa"
-      -e 's!mecab-python3==[^"]*!mecab-python3!' \
+      "mecab-python3"
-      -e 's!numba==[^"]*!numba!' \
+      "numba"
-      -e 's!numpy==[^"]*!numpy!' \
+      "numpy"
-      -e 's!umap-learn==[^"]*!umap-learn!'
+      "umap-learn"
+      "torch"
+    ];
+  in ''
+    sed -r -i \
+      ${lib.concatStringsSep "\n" (map (package:
+        ''-e 's/${package}.*[<>=]+.*/${package}/g' \''
+      ) relaxedConstraints)}
+    requirements.txt
   '';
 
   nativeBuildInputs = with python3.pkgs; [
@@ -43,8 +52,8 @@ python3.pkgs.buildPythonApplication rec {
     coqpit
     flask
     fsspec
-    gruut
     gdown
+    gruut
     inflect
     jieba
     librosa
@@ -54,15 +63,17 @@ python3.pkgs.buildPythonApplication rec {
     pandas
     pypinyin
     pysbd
-    pytorch
+    pytorch-bin
     pyworld
     scipy
     soundfile
     tensorboardx
     tensorflow
+    torchaudio-bin
     tqdm
     umap-learn
     unidic-lite
+    webrtcvad
   ];
 
   postInstall = ''