experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update TF 2.0 package to require Estimator 2.0 package.

Browse Source 
PiperOrigin-RevId: 225321405
This commit is contained in:
Anna R 2018-12-13 00:02:41 -08:00 committed by TensorFlower Gardener
parent b99d914cfc
commit 6beb8734fb
1 changed files with 2 additions and 0 deletions

2
tensorflow/tools/pip_package/setup.py
View File

@ -89,6 +89,8 @@ if 'tf_nightly' in project_name:
REQUIRED_PACKAGES[i] = 'tb-nightly >= 1.13.0a0, < 1.14.0a0'
if 'tensorflow_estimator' in pkg:
REQUIRED_PACKAGES[i] = 'tf-estimator-nightly'
if 'tensorflow_estimator' in pkg and '2.0' in project_name:
REQUIRED_PACKAGES[i] = 'tensorflow-estimator-2.0-preview'
# weakref.finalize and enum were introduced in Python 3.4
if sys.version_info < (3, 4):