Update TF version requirements

PiperOrigin-RevId: 259451160
tensorflow · Jul 23, 2019 · 76002bd · 76002bd
1 parent 907e7b2
commit 76002bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/oss_scripts/pip_package/setup.py b/oss_scripts/pip_package/setup.py
@@ -73,11 +73,11 @@ def finalize_options(self):
     cmdclass={'install': InstallPlatlib},
     distclass=BinaryDistribution,
     install_requires=[
-        'tensorflow==2.0',
+        'tensorflow>=2.0.0b0, <2.1',
     ],
     extras_require={
         'tensorflow_gpu': [
-            'tensorflow-gpu==2.0',
+            'tensorflow-gpu>=2.0.0b0, <2.1',
         ],
         'tests': [
             'absl-py',