diff --git a/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Dataflow.json b/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Dataflow.json index b60f5c4cc3c8..bb5da04014ec 100644 --- a/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Dataflow.json +++ b/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Dataflow.json @@ -1,4 +1,4 @@ { "comment": "Modify this file in a trivial way to cause this test suite to run", - "modification": 0 + "modification": 15 } diff --git a/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Direct.json b/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Direct.json index 99a8fc8ff6d5..bb5da04014ec 100644 --- a/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Direct.json +++ b/.github/trigger_files/beam_PostCommit_Python_Xlang_Gcp_Direct.json @@ -1,4 +1,4 @@ { "comment": "Modify this file in a trivial way to cause this test suite to run", - "modification": 14 + "modification": 15 } diff --git a/sdks/python/apache_beam/io/gcp/tests/xlang_spannerio_it_test.py b/sdks/python/apache_beam/io/gcp/tests/xlang_spannerio_it_test.py index b5d5304245c4..3fe56540847e 100644 --- a/sdks/python/apache_beam/io/gcp/tests/xlang_spannerio_it_test.py +++ b/sdks/python/apache_beam/io/gcp/tests/xlang_spannerio_it_test.py @@ -103,6 +103,11 @@ def setUpClass(cls): ) pipeline = TestPipeline(is_integration_test=True) + + runner_name = type(pipeline.runner).__name__ + if 'DataflowRunner' in runner_name: + pytest.skip("Spanner emulator not compatible with dataflow runner.") + argv = pipeline.get_full_options_as_args() known_args, _ = parser.parse_known_args(argv) diff --git a/sdks/python/apache_beam/ml/rag/ingestion/spanner_it_test.py b/sdks/python/apache_beam/ml/rag/ingestion/spanner_it_test.py index ab9a982a81f7..c371d6fd96b4 100644 --- a/sdks/python/apache_beam/ml/rag/ingestion/spanner_it_test.py +++ b/sdks/python/apache_beam/ml/rag/ingestion/spanner_it_test.py @@ -138,6 +138,11 @@ class SpannerVectorWriterTest(unittest.TestCase): @classmethod def setUpClass(cls): """Set up Spanner emulator for all tests.""" + pipeline = TestPipeline(is_integration_test=True) + runner_name = type(pipeline.runner).__name__ + if 'DataflowRunner' in runner_name: + pytest.skip("Spanner emulator not compatible with dataflow runner.") + cls.project_id = 'test-project' cls.instance_id = 'test-instance' cls.table_name = 'embeddings'