Prepare for TFMA 0.15.1 Release

PiperOrigin-RevId: 276153476

Author: tf-model-analysis-team

README.md

@@ -62,6 +62,7 @@ combinations may also work.
 |tensorflow-model-analysis                                                           |tensorflow    |apache-beam[gcp]|
 |------------------------------------------------------------------------------------|--------------|----------------|
 |[GitHub master](https://github.com/tensorflow/model-analysis/blob/master/RELEASE.md)|nightly (1.x/2.x) |2.16.0          |
+|[0.15.1](https://github.com/tensorflow/model-analysis/blob/v0.15.1/RELEASE.md)      |1.15 / 2.0    |2.16.0          |
 |[0.15.0](https://github.com/tensorflow/model-analysis/blob/v0.15.0/RELEASE.md)      |1.15          |2.16.0          |
 |[0.14.0](https://github.com/tensorflow/model-analysis/blob/v0.14.0/RELEASE.md)      |1.14          |2.14.0          |
 |[0.13.1](https://github.com/tensorflow/model-analysis/blob/v0.13.1/RELEASE.md)      |1.13          |2.11.0          |

RELEASE.md

@@ -1,4 +1,4 @@
-# Current version (not yet released; still in development)
+# Release 0.15.1
 
 ## Major Features and Improvements
 
@@ -21,6 +21,14 @@
 
 ## Bug fixes and other changes
 
+*   Depends on `tensorflow>=1.15,<3.0`.
+    *   Starting from 1.15, package `tensorflow` comes with GPU support. Users
+        won't need to choose between `tensorflow` and `tensorflow-gpu`.
+    *   Caveat: `tensorflow` 2.0.0 is an exception and does not have GPU
+        support. If `tensorflow-gpu` 2.0.0 is installed before installing
+        `tensorflow_model_analysis`, it will be replaced with `tensorflow`
+        2.0.0. Re-install `tensorflow-gpu` 2.0.0 if needed.
+
 ## Breaking changes
 
 ## Deprecations
@@ -214,7 +222,8 @@
 *   Depends on `numpy>=1.14.5,<2`.
 *   Depends on `scipy==0.19.1`.
 *   Depends on `protobuf==3.7.0rc2`.
-*   Chicago Taxi example is moved to tfx repo (https://github.com/tensorflow/tfx/tree/master/tfx/examples/chicago_taxi)
+*   Chicago Taxi example is moved to tfx repo
+    (https://github.com/tensorflow/tfx/tree/master/tfx/examples/chicago_taxi)
 
 ## Breaking changes
 

g3doc/install.md

@@ -58,6 +58,7 @@ combinations may also work.
 tensorflow-model-analysis                                                            | tensorflow        | apache-beam[gcp]
 ------------------------------------------------------------------------------------ | ----------------- | ----------------
 [GitHub master](https://github.com/tensorflow/model-analysis/blob/master/RELEASE.md) | nightly (1.x/2.x) | 2.16.0
+[0.15.1](https://github.com/tensorflow/model-analysis/blob/v0.15.1/RELEASE.md)       | 1.15 / 2.0        | 2.16.0
 [0.15.0](https://github.com/tensorflow/model-analysis/blob/v0.15.0/RELEASE.md)       | 1.15              | 2.16.0
 [0.14.0](https://github.com/tensorflow/model-analysis/blob/v0.14.0/RELEASE.md)       | 1.14              | 2.14.0
 [0.13.1](https://github.com/tensorflow/model-analysis/blob/v0.13.1/RELEASE.md)       | 1.13              | 2.11.0

setup.py

@@ -264,10 +264,8 @@ def run(self):
         # TODO(b/126957988): Stop pinning scipy when possible.
         'scipy==1.1.0',
         'six>=1.12,<2',
-        # TODO(xinzha): Uncomment this once TF can automatically select between
-        # CPU and GPU installation.
-        # 'tensorflow>=1.15,<3',
-        'tfx-bsl>=0.15.0,<0.17'
+        'tensorflow>=1.15,<3',
+        'tfx-bsl>=0.15,<0.16'
     ],
     'python_requires': '>=2.7,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,<4',
     'packages': find_packages(),

tensorflow_model_analysis/version.py

@@ -15,4 +15,4 @@
 
 # Version string for this release of TFMA.
 # Note that setup.py reads and uses this version.
-VERSION_STRING = '0.16.0dev'
+VERSION_STRING = '0.15.1'