Merge branch 'main' into mor--debugging-terminal-integration-customjo…

…b-hp-customtrainingjob
googleapis · Oct 4, 2021 · 6518fb3 · 6518fb3
2 parents 84304ed + ac18182
commit 6518fb3
Show file tree

Hide file tree

Showing 12 changed files with 47 additions and 2 deletions.
diff --git a/.kokoro/continuous/common.cfg b/.kokoro/continuous/common.cfg
@@ -25,3 +25,7 @@ env_vars: {
     key: "TRAMPOLINE_BUILD_FILE"
     value: "github/python-aiplatform/.kokoro/build.sh"
 }
+env_vars: {
+    key: "BUILD_SPECIFIC_GCLOUD_PROJECT"
+    value: "ucaip-sample-tests"
+}
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,23 @@
 # Changelog
 
+## [1.5.0](https://www.github.com/googleapis/python-aiplatform/compare/v1.4.3...v1.5.0) (2021-09-30)
+
+
+### Features
+
+* Add data plane code snippets for feature store service ([#713](https://www.github.com/googleapis/python-aiplatform/issues/713)) ([e3ea683](https://www.github.com/googleapis/python-aiplatform/commit/e3ea683bf754832340853a15bdb0a0662500a70f))
+* add flaky test diagnostic script ([#734](https://www.github.com/googleapis/python-aiplatform/issues/734)) ([09e48de](https://www.github.com/googleapis/python-aiplatform/commit/09e48de8b79fb5d601169663c9a8e1c33883f1cf))
+* add vizier service to aiplatform v1 BUILD.bazel ([#731](https://www.github.com/googleapis/python-aiplatform/issues/731)) ([1a580ae](https://www.github.com/googleapis/python-aiplatform/commit/1a580aec158b5e25b94f27a6a9daa3943124c485))
+* code snippets for feature store control plane ([#709](https://www.github.com/googleapis/python-aiplatform/issues/709)) ([8e06ced](https://www.github.com/googleapis/python-aiplatform/commit/8e06ced83ed2cc480d869318c4debef9c28ad214))
+* Updating the Tensorboard uploader to use the new batch write API so it runs more efficiently ([#710](https://www.github.com/googleapis/python-aiplatform/issues/710)) ([9d1b01a](https://www.github.com/googleapis/python-aiplatform/commit/9d1b01a91dc077bfe8edf023216b65b826d67d5f))
+
+
+### Bug Fixes
+
+* [#677](https://www.github.com/googleapis/python-aiplatform/issues/677) ([#728](https://www.github.com/googleapis/python-aiplatform/issues/728)) ([7f548e4](https://www.github.com/googleapis/python-aiplatform/commit/7f548e4b5322055a3c2befcdc9d4eef1bc2278ca))
+* **PipelineJob:** use name as output only field ([#719](https://www.github.com/googleapis/python-aiplatform/issues/719)) ([1c84464](https://www.github.com/googleapis/python-aiplatform/commit/1c84464e3130f9db81cd341306b334f9a490587f))
+* use the project id from BQ dataset instead of the default project id ([#717](https://www.github.com/googleapis/python-aiplatform/issues/717)) ([e87a255](https://www.github.com/googleapis/python-aiplatform/commit/e87a255705a5d04ade79f12c706dc842c0228118))
+
 ### [1.4.3](https://www.github.com/googleapis/python-aiplatform/compare/v1.4.2...v1.4.3) (2021-09-17)
 
 

diff --git a/README.rst b/README.rst
@@ -1,7 +1,7 @@
 Vertex SDK for Python
 =================================================
 
-|GA| |pypi| |versions|
+|GA| |pypi| |versions| |unit-tests| |system-tests| |sample-tests|
 
 
 `Vertex AI`_: Google Vertex AI is an integrated suite of machine learning tools and services for building and using ML models with AutoML or custom code. It offers both novices and experts the best workbench for the entire machine learning development lifecycle.
@@ -15,6 +15,12 @@ Vertex SDK for Python
    :target: https://pypi.org/project/google-cloud-aiplatform/
 .. |versions| image:: https://img.shields.io/pypi/pyversions/google-cloud-aiplatform.svg
    :target: https://pypi.org/project/google-cloud-aiplatform/
+.. |unit-tests| image:: https://storage.googleapis.com/cloud-devrel-public/python-aiplatform/badges/sdk-unit-tests.svg
+   :target: https://storage.googleapis.com/cloud-devrel-public/python-aiplatform/badges/sdk-unit-tests.html
+.. |system-tests| image:: https://storage.googleapis.com/cloud-devrel-public/python-aiplatform/badges/sdk-system-tests.svg
+   :target: https://storage.googleapis.com/cloud-devrel-public/python-aiplatform/badges/sdk-system-tests.html
+.. |sample-tests| image:: https://storage.googleapis.com/cloud-devrel-public/python-aiplatform/badges/sdk-sample-tests.svg
+   :target: https://storage.googleapis.com/cloud-devrel-public/python-aiplatform/badges/sdk-sample-tests.html
 .. _Vertex AI: https://cloud.google.com/vertex-ai/docs
 .. _Client Library Documentation: https://googleapis.dev/python/aiplatform/latest
 .. _Product Documentation:  https://cloud.google.com/vertex-ai/docs

diff --git a/samples/snippets/pipeline_service/create_training_pipeline_sample_test.py b/samples/snippets/pipeline_service/create_training_pipeline_sample_test.py
@@ -34,6 +34,8 @@ def teardown(teardown_training_pipeline):
 # Training AutoML Vision Model
 def test_ucaip_generated_create_training_pipeline_sample(capsys, shared_state):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_sample.create_training_pipeline_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/.../snippets/pipeline_service/create_training_pipeline_tabular_classification_sample_test.py b/.../snippets/pipeline_service/create_training_pipeline_tabular_classification_sample_test.py
@@ -35,6 +35,8 @@ def teardown(teardown_training_pipeline):
 @pytest.mark.skip(reason="https://github.com/googleapis/java-aiplatform/issues/420")
 def test_ucaip_generated_create_training_pipeline_sample(capsys, shared_state):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_tabular_classification_sample.create_training_pipeline_tabular_classification_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/...les/snippets/pipeline_service/create_training_pipeline_tabular_forecasting_sample_test.py b/...les/snippets/pipeline_service/create_training_pipeline_tabular_forecasting_sample_test.py
@@ -67,6 +67,8 @@ def teardown(shared_state):
 @pytest.mark.skip(reason="https://github.com/googleapis/java-aiplatform/issues/420")
 def test_ucaip_generated_create_training_pipeline_sample(capsys, shared_state):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_tabular_forecasting_sample.create_training_pipeline_tabular_forecasting_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/.../snippets/pipeline_service/create_training_pipeline_text_entity_extraction_sample_test.py b/.../snippets/pipeline_service/create_training_pipeline_text_entity_extraction_sample_test.py
@@ -35,6 +35,8 @@ def test_ucaip_generated_create_training_pipeline_text_entity_extraction_sample(
     capsys, shared_state
 ):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_text_entity_extraction_sample.create_training_pipeline_text_entity_extraction_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/...snippets/pipeline_service/create_training_pipeline_text_sentiment_analysis_sample_test.py b/...snippets/pipeline_service/create_training_pipeline_text_sentiment_analysis_sample_test.py
@@ -36,6 +36,8 @@ def test_ucaip_generated_create_training_pipeline_text_sentiment_analysis_sample
     capsys, shared_state
 ):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_text_sentiment_analysis_sample.create_training_pipeline_text_sentiment_analysis_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/...nippets/pipeline_service/create_training_pipeline_video_action_recognition_sample_test.py b/...nippets/pipeline_service/create_training_pipeline_video_action_recognition_sample_test.py
@@ -40,6 +40,9 @@ def teardown(teardown_training_pipeline):
 @pytest.mark.skip(reason="https://github.com/googleapis/java-aiplatform/issues/420")
 # Training AutoML Vision Model
 def test_create_training_pipeline_video_action_recognition_sample(capsys, shared_state):
+
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_video_action_recognition_sample.create_training_pipeline_video_action_recognition_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/...es/snippets/pipeline_service/create_training_pipeline_video_classification_sample_test.py b/...es/snippets/pipeline_service/create_training_pipeline_video_classification_sample_test.py
@@ -36,6 +36,8 @@ def test_ucaip_generated_create_training_pipeline_video_classification_sample(
     capsys, shared_state
 ):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_video_classification_sample.create_training_pipeline_video_classification_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/...s/snippets/pipeline_service/create_training_pipeline_video_object_tracking_sample_test.py b/...s/snippets/pipeline_service/create_training_pipeline_video_object_tracking_sample_test.py
@@ -35,6 +35,8 @@ def test_ucaip_generated_create_training_pipeline_video_object_tracking_sample(
     capsys, shared_state
 ):
 
+    shared_state["cancel_batch_prediction_job_timeout"] = 300
+
     create_training_pipeline_video_object_tracking_sample.create_training_pipeline_video_object_tracking_sample(
         project=PROJECT_ID,
         display_name=DISPLAY_NAME,

diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
 import setuptools  # type: ignore
 
 name = "google-cloud-aiplatform"
-version = "1.4.3"
+version = "1.5.0"
 description = "Cloud AI Platform API client library"
 
 package_root = os.path.abspath(os.path.dirname(__file__))