Updated job_state for Vertex (#4416)

PiperOrigin-RevId: 405761816
(cherry picked from commit a0a21df)

Co-authored-by: tfx-team <tensorflow-extended-nonhuman@googlegroups.com>

README.md

@@ -63,6 +63,7 @@ but other *untested* combinations may also work.
 tfx                                                                       | apache-beam[gcp] | ml-metadata | pyarrow | tensorflow        | tensorflow-data-validation | tensorflow-metadata | tensorflow-model-analysis | tensorflow-serving-api | tensorflow-transform | tfx-bsl
 ------------------------------------------------------------------------- | ---------------- | ----------- | ------- | ----------------- | -------------------------- | ------------------- | ------------------------- | ---------------------- | -------------------- | -------
 [GitHub master](https://github.com/tensorflow/tfx/blob/master/RELEASE.md) | 2.32.0           | 1.3.0       | 2.0.0   | nightly (1.x/2.x) | 1.3.0                      | 1.2.0               | 0.34.1                    | 2.6.0                  | 1.3.0                | 1.3.0
+1.3.2                                                                     | 2.32.0           | 1.3.0       | 2.0.0   | 1.15.0 / 2.6.0    | 1.3.0                      | 1.2.0               | 0.34.1                    | 2.6.0                  | 1.3.0                | 1.3.0
 1.3.1                                                                     | 2.32.0           | 1.3.0       | 2.0.0   | 1.15.0 / 2.6.0    | 1.3.0                      | 1.2.0               | 0.34.1                    | 2.6.0                  | 1.3.0                | 1.3.0
 1.3.0                                                                     | 2.32.0           | 1.3.0       | 2.0.0   | 1.15.0 / 2.6.0    | 1.3.0                      | 1.2.0               | 0.34.1                    | 2.6.0                  | 1.3.0                | 1.3.0
 1.2.0                                                                     | 2.31.0           | 1.2.0       | 2.0.0   | 1.15.0 / 2.5.0    | 1.2.0                      | 1.2.0               | 0.33.0                    | 2.5.1                  | 1.2.0                | 1.2.0

RELEASE.md

@@ -1,4 +1,4 @@
-# Version 1.3.1
+# Version 1.3.2
 
 ## Major Features and Improvements
 
@@ -18,7 +18,7 @@
 
 ## Bug Fixes and Other Changes
 
-*  Fixed Vertex Pusher by passing enable_vertex flag for deploying model.
+*  Fixed endless waiting for Vertex Trainer.
 
 ## Documentation Updates
 

tfx/extensions/google_cloud_ai_platform/runner_test.py

@@ -25,10 +25,9 @@
 from google.cloud.aiplatform import initializer
 from google.cloud.aiplatform.compat.types import endpoint
 from google.cloud.aiplatform_v1.services.endpoint_service import (
-    client as endpoint_service_client,
-)
-from google.cloud.aiplatform_v1beta1.types.custom_job import CustomJob
-from google.cloud.aiplatform_v1beta1.types.job_state import JobState
+    client as endpoint_service_client)
+from google.cloud.aiplatform_v1.types.custom_job import CustomJob
+from google.cloud.aiplatform_v1.types.job_state import JobState
 from googleapiclient import errors
 import httplib2
 import tensorflow as tf

tfx/extensions/google_cloud_ai_platform/training_clients.py

@@ -21,8 +21,8 @@
 
 from absl import logging
 from google.cloud.aiplatform import gapic
-from google.cloud.aiplatform_v1beta1.types.custom_job import CustomJob
-from google.cloud.aiplatform_v1beta1.types.job_state import JobState
+from google.cloud.aiplatform_v1.types.custom_job import CustomJob
+from google.cloud.aiplatform_v1.types.job_state import JobState
 from googleapiclient import discovery
 
 from tfx import types

tfx/version.py

@@ -14,4 +14,4 @@
 """Contains the version string of TFX."""
 
 # Note that setup.py uses this version.
-__version__ = '1.3.1'
+__version__ = '1.3.2'