diff --git a/pyproject.toml b/pyproject.toml
index 943d3ac..4ab5859 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,6 +55,7 @@ test = [
   "pyyaml>=6.0.3",
   "pyzmq>=27.1.0",
   "soundfile>=0.13.1",
+  "tf-keras>=2.20.1",
 ]
 
 [tool.pytest.ini_options]
diff --git a/uv.lock b/uv.lock
index ff4a8be..b87cc0b 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1555,6 +1555,7 @@ test = [
     { name = "pyyaml" },
     { name = "pyzmq" },
     { name = "soundfile" },
+    { name = "tf-keras" },
 ]
 
 [package.metadata]
@@ -1609,6 +1610,7 @@ test = [
     { name = "pyyaml", specifier = ">=6.0.3" },
     { name = "pyzmq", specifier = ">=27.1.0" },
     { name = "soundfile", specifier = ">=0.13.1" },
+    { name = "tf-keras", specifier = ">=2.20.1" },
 ]
 
 [[package]]