[beam] branch master updated: Fix Vertex AI postcommit suite to run the tests only once. (#28571)

[tvalentyn]

This is an automated email from the ASF dual-hosted git repository.

tvalentyn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 27f589e34f1 Fix Vertex AI postcommit suite to run the tests only once. 
(#28571)
27f589e34f1 is described below

commit 27f589e34f1b17f7ea288af03e255f83c2462e03
Author: tvalentyn <tvalen...@users.noreply.github.com>
AuthorDate: Mon Sep 25 17:17:29 2023 -0700

    Fix Vertex AI postcommit suite to run the tests only once. (#28571)
---
 sdks/python/apache_beam/ml/inference/vertex_ai_inference_it_test.py | 3 +--
 sdks/python/pytest.ini                                              | 2 +-
 sdks/python/test-suites/dataflow/common.gradle                      | 2 +-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git 
a/sdks/python/apache_beam/ml/inference/vertex_ai_inference_it_test.py 
b/sdks/python/apache_beam/ml/inference/vertex_ai_inference_it_test.py
index 8ba315ba00c..168ab031abb 100644
--- a/sdks/python/apache_beam/ml/inference/vertex_ai_inference_it_test.py
+++ b/sdks/python/apache_beam/ml/inference/vertex_ai_inference_it_test.py
@@ -46,8 +46,7 @@ _SUBNETWORK = 
"https://www.googleapis.com/compute/v1/projects/apache-beam-testin
 
 
 class VertexAIInference(unittest.TestCase):
-  @pytest.mark.uses_vertex_ai
-  @pytest.mark.it_postcommit
+  @pytest.mark.vertex_ai_postcommit
   def test_vertex_ai_run_flower_image_classification(self):
     output_file = '/'.join([_OUTPUT_DIR, str(uuid.uuid4()), 'output.txt'])
 
diff --git a/sdks/python/pytest.ini b/sdks/python/pytest.ini
index ec8d9fb3900..d1d69f79835 100644
--- a/sdks/python/pytest.ini
+++ b/sdks/python/pytest.ini
@@ -60,7 +60,7 @@ markers =
     uses_onnx: tests that utilizes onnx in some way.
     uses_tf: tests that utilize tensorflow.
     uses_transformers: tests that utilize transformers in some way.
-    uses_vertex_ai: tests that utilize vertex ai in some way.
+    vertex_ai_postcommit: vertex ai postcommits that need additional deps.
 
 # Default timeout intended for unit tests.
 # If certain tests need a different value, please see the docs on how to
diff --git a/sdks/python/test-suites/dataflow/common.gradle 
b/sdks/python/test-suites/dataflow/common.gradle
index 8b8a5680899..f907a728c0b 100644
--- a/sdks/python/test-suites/dataflow/common.gradle
+++ b/sdks/python/test-suites/dataflow/common.gradle
@@ -417,7 +417,7 @@ task vertexAIInferenceTest {
       def argMap = [
           "test_opts": testOpts,
           "suite": "VertexAITests-df-py${pythonVersionSuffix}",
-          "collect": "uses_vertex_ai and it_postcommit" ,
+          "collect": "vertex_ai_postcommit" ,
           "runner": "TestDataflowRunner",
           "requirements_file": "$requirementsFile"
       ]