No public description

PiperOrigin-RevId: 638676207 Change-Id: I3a8ec0b37adc9da7f1ce878118c5635e55fffd58
google-deepmind · May 30, 2024 · 4489384 · 4489384
1 parent 410e70e
commit 4489384
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -9,11 +9,12 @@ goal.
 
 Currently we use:
 
-*   flax for modules
-*   optax for optimization
-*   orbax for checkpointing
-*   pygrain for data
-*   ConfigDict for hyper-parameters.
+*   [flax](https://github.com/google/flax) for modules
+*   [optax](https://github.com/google-deepmind/optax) for optimization
+*   [orbax](https://github.com/google/orbax) for checkpointing
+*   [tfds](https://github.com/tensorflow/datasets) for data
+*   [pygrain](https://github.com/google/grain) for data loading
+*   [ConfigDict](https://github.com/google/ml_collections) for hyper-parameters.
 
 Not currently supported or left out for simplicity:
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -60,8 +60,7 @@ dependencies = [
     "optax>=0.2.2",
     "orbax>=0.1.7",
     "sentencepiece>=0.2.0",
-    # TODO: Temporary fix while waiting for TFDS to be released.
-    "tfds-nightly",
+    "tensorflow_datasets>=4.9.5",
     "tensorflow>=2.16.1",
 ]