[beam] branch master updated: Add extra marks to VertexAI IT tests (#28534)

[tv...@apache.org]

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 a1ec0552a34 Add extra marks to VertexAI IT tests (#28534)
a1ec0552a34 is described below

commit a1ec0552a340aa99a6d0aadc77f49caf8acab110
Author: Jack McCluskey <34...@users.noreply.github.com>
AuthorDate: Tue Sep 19 18:18:12 2023 -0400

    Add extra marks to VertexAI IT tests (#28534)
---
 sdks/python/apache_beam/ml/inference/vertex_ai_inference_it_test.py | 1 +
 sdks/python/test-suites/dataflow/common.gradle                      | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

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 4ef42fb10a7..8ba315ba00c 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
@@ -26,6 +26,7 @@ import pytest
 from apache_beam.io.filesystems import FileSystems
 from apache_beam.testing.test_pipeline import TestPipeline
 
+# pylint: disable=ungrouped-imports
 try:
   from apache_beam.examples.inference import vertex_ai_image_classification
   from apache_beam.examples.inference import vertex_ai_llm_text_classification
diff --git a/sdks/python/test-suites/dataflow/common.gradle b/sdks/python/test-suites/dataflow/common.gradle
index eee6e9d2188..8b8a5680899 100644
--- a/sdks/python/test-suites/dataflow/common.gradle
+++ b/sdks/python/test-suites/dataflow/common.gradle
@@ -418,7 +418,8 @@ task vertexAIInferenceTest {
           "test_opts": testOpts,
           "suite": "VertexAITests-df-py${pythonVersionSuffix}",
           "collect": "uses_vertex_ai and it_postcommit" ,
-          "runner": "TestDataflowRunner"
+          "runner": "TestDataflowRunner",
+          "requirements_file": "$requirementsFile"
       ]
       def cmdArgs = mapToArgString(argMap)
       exec {