From 4489384bf90f88d284f0380558cc39a72d2ef822 Mon Sep 17 00:00:00 2001
From: Pierre Marcenac <pierremarcenac@google.com>
Date: Thu, 30 May 2024 16:16:23 +0000
Subject: [PATCH] No public description

PiperOrigin-RevId: 638676207
Change-Id: I3a8ec0b37adc9da7f1ce878118c5635e55fffd58
---
 README.md      | 11 ++++++-----
 pyproject.toml |  3 +--
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/README.md b/README.md
index f555346..929345f 100644
--- 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
index 020d60d..c29a72a 100644
--- 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",
 ]