Add tensorflow-datasets to doc requirements

PiperOrigin-RevId: 589532101
google-deepmind · Dec 10, 2023 · b8959c6 · b8959c6
1 parent d767626
commit b8959c6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -70,7 +70,8 @@ docs = [
     "dm-haiku==0.0.11",
     "sphinx-gallery==0.14.0",
     "sphinx-collections==0.0.1",
-    "tensorflow>=2.4.0"
+    "tensorflow>=2.4.0",
+    "tensorflow-datasets>=4.2.0",
 ]
 
 dp-accounting = [