diff --git a/pkgs/development/python-modules/torch-audiomentations/default.nix b/pkgs/development/python-modules/torch-audiomentations/default.nix
index ca01c81ced29..dbfa7cbbb5e2 100644
--- a/pkgs/development/python-modules/torch-audiomentations/default.nix
+++ b/pkgs/development/python-modules/torch-audiomentations/default.nix
@@ -16,6 +16,7 @@
   # tests
   pytest-cov-stub,
   pytestCheckHook,
+  pyyaml,
 }:
 
 buildPythonPackage (finalAttrs: {
@@ -45,6 +46,7 @@ buildPythonPackage (finalAttrs: {
   nativeCheckInputs = [
     pytest-cov-stub
     pytestCheckHook
+    pyyaml
   ];
 
   pythonImportsCheck = [ "torch_audiomentations" ];