ping tfds to a specific version

PiperOrigin-RevId: 572314851
google · Oct 10, 2023 · 9e85c08 · 9e85c08
1 parent bae5d7e
commit 9e85c08
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -54,8 +54,10 @@
         'pyglove',
         'sentencepiece',
         'tensorflow-text',
-        'tfds-nightly',
-        # remove this when sentencepiece_model_pb2 is re-generated in the
+        # Ping to a specific version to avoid endless backtracking during
+        # pip dependency resolution.
+        'tfds-nightly==4.9.2.dev202308090034',
+        # Remove this when sentencepiece_model_pb2 is re-generated in the
         # sentencepiece package.
         'protobuf<=3.20.3',
     ],