chore: resolve flaky vertexai system tests

jaycee-li · copybara-github · commit 1e4a4ec5a31d · 2023-10-30T11:48:11.000-07:00
PiperOrigin-RevId: 577908201
diff --git a/tests/system/vertexai/test_bigframes_sklearn.py b/tests/system/vertexai/test_bigframes_sklearn.py
@@ -60,6 +60,8 @@
     if os.environ.get("KOKORO_GIT_COMMIT")
     else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",
 )
+# To avoid flaky test due to autolog enabled in parallel tests
+@mock.patch.object(vertexai.preview.global_config, "autolog", False)
 @pytest.mark.usefixtures(
     "prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"
 )
diff --git a/tests/system/vertexai/test_bigframes_tensorflow.py b/tests/system/vertexai/test_bigframes_tensorflow.py
@@ -56,6 +56,8 @@
     if os.environ.get("KOKORO_GIT_COMMIT")
     else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",
 )
+# To avoid flaky test due to autolog enabled in parallel tests
+@mock.patch.object(vertexai.preview.global_config, "autolog", False)
 @pytest.mark.usefixtures(
     "prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"
 )
diff --git a/tests/system/vertexai/test_pytorch.py b/tests/system/vertexai/test_pytorch.py
@@ -48,6 +48,8 @@
     if os.environ.get("KOKORO_GIT_COMMIT")
     else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",
 )
+# To avoid flaky test due to autolog enabled in parallel tests
+@mock.patch.object(vertexai.preview.global_config, "autolog", False)
 @pytest.mark.usefixtures(
     "prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"
 )
diff --git a/tests/system/vertexai/test_sklearn.py b/tests/system/vertexai/test_sklearn.py
@@ -54,6 +54,8 @@
     if os.environ.get("KOKORO_GIT_COMMIT")
     else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",
 )
+# To avoid flaky test due to autolog enabled in parallel tests
+@mock.patch.object(vertexai.preview.global_config, "autolog", False)
 @pytest.mark.usefixtures(
     "prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"
 )
diff --git a/tests/system/vertexai/test_tensorflow.py b/tests/system/vertexai/test_tensorflow.py
@@ -53,6 +53,8 @@
     if os.environ.get("KOKORO_GIT_COMMIT")
     else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",
 )
+# To avoid flaky test due to autolog enabled in parallel tests
+@mock.patch.object(vertexai.preview.global_config, "autolog", False)
 @pytest.mark.usefixtures(
     "prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"
 )

Original file line number	Diff line number	Diff line change
`@@ -60,6 +60,8 @@`
`60`	`60`	`if os.environ.get("KOKORO_GIT_COMMIT")`
`61`	`61`	`else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",`
`62`	`62`	`)`
	`63`	`+# To avoid flaky test due to autolog enabled in parallel tests`
	`64`	`+@mock.patch.object(vertexai.preview.global_config, "autolog", False)`
`63`	`65`	`@pytest.mark.usefixtures(`
`64`	`66`	`"prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"`
`65`	`67`	`)`
Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,8 @@`
`56`	`56`	`if os.environ.get("KOKORO_GIT_COMMIT")`
`57`	`57`	`else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",`
`58`	`58`	`)`
	`59`	`+# To avoid flaky test due to autolog enabled in parallel tests`
	`60`	`+@mock.patch.object(vertexai.preview.global_config, "autolog", False)`
`59`	`61`	`@pytest.mark.usefixtures(`
`60`	`62`	`"prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"`
`61`	`63`	`)`
Original file line number	Diff line number	Diff line change
`@@ -48,6 +48,8 @@`
`48`	`48`	`if os.environ.get("KOKORO_GIT_COMMIT")`
`49`	`49`	`else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",`
`50`	`50`	`)`
	`51`	`+# To avoid flaky test due to autolog enabled in parallel tests`
	`52`	`+@mock.patch.object(vertexai.preview.global_config, "autolog", False)`
`51`	`53`	`@pytest.mark.usefixtures(`
`52`	`54`	`"prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"`
`53`	`55`	`)`
Original file line number	Diff line number	Diff line change
`@@ -54,6 +54,8 @@`
`54`	`54`	`if os.environ.get("KOKORO_GIT_COMMIT")`
`55`	`55`	`else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",`
`56`	`56`	`)`
	`57`	`+# To avoid flaky test due to autolog enabled in parallel tests`
	`58`	`+@mock.patch.object(vertexai.preview.global_config, "autolog", False)`
`57`	`59`	`@pytest.mark.usefixtures(`
`58`	`60`	`"prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"`
`59`	`61`	`)`
Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,8 @@`
`53`	`53`	`if os.environ.get("KOKORO_GIT_COMMIT")`
`54`	`54`	`else "google-cloud-aiplatform[preview,autologging] @ git+https://github.com/googleapis/python-aiplatform.git@main",`
`55`	`55`	`)`
	`56`	`+# To avoid flaky test due to autolog enabled in parallel tests`
	`57`	`+@mock.patch.object(vertexai.preview.global_config, "autolog", False)`
`56`	`58`	`@pytest.mark.usefixtures(`
`57`	`59`	`"prepare_staging_bucket", "delete_staging_bucket", "tear_down_resources"`
`58`	`60`	`)`