From 36af4adf9d7dbc862cdf4fc2dbab0295ac28da07 Mon Sep 17 00:00:00 2001
From: Martin Weinelt <hexa@darmstadt.ccc.de>
Date: Thu, 29 Jun 2023 02:11:00 +0200
Subject: [PATCH 1/3] python310Packages.encodec: init at 0.1.1

EnCodec: High Fidelity Neural Audio Compression
---
 .../python-modules/encodec/default.nix        | 43 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 +
 2 files changed, 45 insertions(+)
 create mode 100644 pkgs/development/python-modules/encodec/default.nix

diff --git a/pkgs/development/python-modules/encodec/default.nix b/pkgs/development/python-modules/encodec/default.nix
new file mode 100644
index 000000000000..930878782f60
--- /dev/null
+++ b/pkgs/development/python-modules/encodec/default.nix
@@ -0,0 +1,43 @@
+{ lib
+, buildPythonPackage
+, fetchFromGitHub
+
+# dependencies
+, einops
+, numpy
+, torch-bin
+, torchaudio-bin
+}:
+
+buildPythonPackage rec {
+  pname = "encodec";
+  version = "0.1.1";
+  format = "setuptools";
+
+  src = fetchFromGitHub {
+    owner = "facebookresearch";
+    repo = "encodec";
+    rev = "v${version}";
+    hash = "sha256-+iJZkX1HoyuNFu9VRxMO6aAzNQybkH9lrQJ5Ao9+/CY=";
+  };
+
+  propagatedBuildInputs = [
+    einops
+    numpy
+    torch-bin
+    torchaudio-bin
+  ];
+
+  pythonImportsCheck = [ "encodec" ];
+
+  # requires model data from the internet
+  doCheck = false;
+
+  meta = with lib; {
+    description = "State-of-the-art deep learning based audio codec supporting both mono 24 kHz audio and stereo 48 kHz audio";
+    homepage = "https://github.com/facebookresearch/encodec";
+    changelog = "https://github.com/facebookresearch/encodec/blob/${src.rev}/CHANGELOG.md";
+    license = licenses.mit;
+    maintainers = with maintainers; [ hexa ];
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 71b66cd0b074..66df6cf075e4 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -3260,6 +3260,8 @@ self: super: with self; {
 
   enamlx = callPackage ../development/python-modules/enamlx { };
 
+  encodec = callPackage ../development/python-modules/encodec { };
+
   energyflip-client = callPackage ../development/python-modules/energyflip-client { };
 
   energyflow = callPackage ../development/python-modules/energyflow { };

From cb39db1224796fd17a8e5ca12a0349166178377c Mon Sep 17 00:00:00 2001
From: Martin Weinelt <hexa@darmstadt.ccc.de>
Date: Thu, 29 Jun 2023 03:14:39 +0200
Subject: [PATCH 2/3] python310Packages.wandb: disable timing sensitive test

wandb> ______________________________ test_login_timeout ______________________________
wandb> [gw3] linux -- Python 3.10.12 /nix/store/1r6n7v2wam7gkr18gxccpg7p5ywgw551-python3-3.10.12/bin/python3.10
wandb>
wandb> mock_tty = <function mock_tty.<locals>.setup_fn at 0x7ffe258248b0>
wandb>
wandb>     def test_login_timeout(mock_tty):
wandb>         mock_tty("junk\nmore\n")
wandb>         start_time = time.time()
wandb>         ret = wandb.login(timeout=4)
wandb>         elapsed = time.time() - start_time
wandb> >       assert 2 < elapsed < 15
wandb> E       assert 15.33935809135437 < 15
wandb>
wandb> /build/source/tests/pytest_tests/unit_tests/test_wandb_login.py:78: AssertionError
---
 pkgs/development/python-modules/wandb/default.nix | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/pkgs/development/python-modules/wandb/default.nix b/pkgs/development/python-modules/wandb/default.nix
index aeeece5151a3..9ccb68067b72 100644
--- a/pkgs/development/python-modules/wandb/default.nix
+++ b/pkgs/development/python-modules/wandb/default.nix
@@ -256,9 +256,12 @@ buildPythonPackage rec {
     "tests/pytest_tests/unit_tests/test_lib/test_filesystem.py"
   ];
 
-  # Disable test that fails on darwin due to issue with python3Packages.psutil:
-  # https://github.com/giampaolo/psutil/issues/1219
-  disabledTests = lib.optionals stdenv.isDarwin [
+  disabledTests = [
+    # Timing sensitive
+    "test_login_timeout"
+  ] ++ lib.optionals stdenv.isDarwin [
+    # Disable test that fails on darwin due to issue with python3Packages.psutil:
+    # https://github.com/giampaolo/psutil/issues/1219
     "test_tpu_system_stats"
   ];
 

From cababbbc357f38b70578dcd701bbc49a758fd833 Mon Sep 17 00:00:00 2001
From: Martin Weinelt <hexa@darmstadt.ccc.de>
Date: Thu, 29 Jun 2023 03:26:55 +0200
Subject: [PATCH 3/3] tts: 0.14.3 -> 0.15.0

https://github.com/coqui-ai/TTS/releases/tag/v0.15.0
---
 pkgs/tools/audio/tts/default.nix | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pkgs/tools/audio/tts/default.nix b/pkgs/tools/audio/tts/default.nix
index d4e5a68fa094..f4c1f568bb7f 100644
--- a/pkgs/tools/audio/tts/default.nix
+++ b/pkgs/tools/audio/tts/default.nix
@@ -15,14 +15,14 @@ let
 in
 python.pkgs.buildPythonApplication rec {
   pname = "tts";
-  version = "0.14.3";
+  version = "0.15.0";
   format = "pyproject";
 
   src = fetchFromGitHub {
     owner = "coqui-ai";
     repo = "TTS";
     rev = "refs/tags/v${version}";
-    hash = "sha256-4LojjH6ft9RfftBBFNWmC2pl/YXfgZCXhkZFsELTgCE=";
+    hash = "sha256-pu0MqNXNQfvxo2VHpiEYIz1OvplydCYPKU/NsZD0mJw=";
   };
 
   postPatch = let
@@ -60,6 +60,7 @@ python.pkgs.buildPythonApplication rec {
     bnunicodenormalizer
     coqpit
     einops
+    encodec
     flask
     fsspec
     g2pkk