From 3a54eb949359f83bd7bb10e2eb2b939c5763ce51 Mon Sep 17 00:00:00 2001
From: MichaelBroughton <MichaelBroughton@users.noreply.github.com>
Date: Wed, 5 Aug 2020 16:03:11 -0700
Subject: [PATCH] Cherry pick (#333) onto v0.3.0.

---
 release/setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/release/setup.py b/release/setup.py
index 89ad51f7c..081216243 100644
--- a/release/setup.py
+++ b/release/setup.py
@@ -51,7 +51,11 @@ def finalize_options(self):
 
 
 REQUIRED_PACKAGES = ['cirq == 0.8.0', 'pathos == 0.2.5', 'sympy == 1.4']
-CUR_VERSION = '0.3.0'
+CUR_VERSION = '0.3.1'
+
+# placed as extra to not have required overwrite existing nightly installs if
+# they exist.
+EXTRA_PACKAGES = ['tensorflow == 2.1.0']
 
 
 class BinaryDistribution(Distribution):
@@ -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'])],