This is an automated email from the ASF dual-hosted git repository. damccorm pushed a commit to branch users/damccorm-repro-35816 in repository https://gitbox.apache.org/repos/asf/beam.git
commit a74a424aaff890d5d10192d7d2a4742b67384ad4 Author: Mohamed Awnallah <[email protected]> AuthorDate: Fri Sep 5 14:00:49 2025 +0000 sdks/python: force tests to run only on precommit --- .../apache_beam/ml/rag/enrichment/milvus_search_it_test.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py b/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py index 4dabcafe670..81ceb6b69e7 100644 --- a/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py +++ b/sdks/python/apache_beam/ml/rag/enrichment/milvus_search_it_test.py @@ -574,7 +574,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): expected_chunks = [] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result, @@ -702,7 +702,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): embedding=Embedding(dense_embedding=[0.3, 0.4, 0.5])) ] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result, @@ -807,7 +807,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): embedding=Embedding()) ] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result, @@ -948,7 +948,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): embedding=Embedding(dense_embedding=[0.3, 0.4, 0.5])) ] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result, @@ -1088,7 +1088,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): embedding=Embedding(dense_embedding=[0.3, 0.4, 0.5])) ] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result, @@ -1153,7 +1153,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): sparse_embedding=([1, 2, 3, 4], [0.05, 0.41, 0.05, 0.41]))) ] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result, @@ -1226,7 +1226,7 @@ class TestMilvusSearchEnrichment(unittest.TestCase): embedding=Embedding(dense_embedding=[0.1, 0.2, 0.3])) ] - with TestPipeline(is_integration_test=True) as p: + with TestPipeline() as p: result = (p | beam.Create(test_chunks) | Enrichment(handler)) assert_that( result,
