Update TensorFlow, Reverb, and tensorflow_probability dependencies to…

… use stable versions. PiperOrigin-RevId: 338657521 Change-Id: Iec6adaf6bc306d6ea882b8f29c5d6832942c8cd3
google-deepmind · Oct 23, 2020 · 0bcbd2f · 0bcbd2f
1 parent d241a62
commit 0bcbd2f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -27,13 +27,13 @@
 spec.loader.exec_module(_metadata)
 
 reverb_requirements = [
-    'dm-reverb-nightly==0.1.0.dev20200708',
-    'tf-nightly==2.4.0.dev20200708',
+    'dm-reverb',
+    'tensorflow>=2.3.0',
 ]
 
 tf_requirements = [
-    'tf-nightly==2.4.0.dev20200708',
-    'tfp-nightly==0.12.0.dev20200717',
+    'tensorflow>=2.3.0',
+    'tensorflow_probability',
     'dm-sonnet',
     'trfl',
 ]