Added tf 2.1.0 under extras. (tensorflow#333)

FermiQ · Aug 5, 2020 · 59a658f · 59a658f
1 parent a757648
commit 59a658f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/release/setup.py b/release/setup.py
@@ -51,6 +51,10 @@ def finalize_options(self):
 
 
 REQUIRED_PACKAGES = ['cirq == 0.8.0', 'sympy == 1.5']
+
+# placed as extra to not have required overwrite existing nightly installs if
+# they exist.
+EXTRA_PACKAGES = ['tensorflow == 2.1.0']
 CUR_VERSION = '0.4.0'
 
 
@@ -83,6 +87,7 @@ def has_ext_modules(self):
     url='https://github.com/tensorflow/quantum/',
     packages=find_packages(),
     install_requires=REQUIRED_PACKAGES,
+    extras_require={'extras': EXTRA_PACKAGES},
     # Add in any packaged data.
     include_package_data=True,
     #ext_modules=[Extension('_foo', ['stub.cc'])],