From 23858970fd9a5893fe356bc7262d76027d8bca1b Mon Sep 17 00:00:00 2001 From: Claude Date: Thu, 19 Dec 2024 20:58:00 -0500 Subject: [PATCH] Fix bigquery streaming test. --- sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py b/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py index 634d9c16fdc..f125fedf8c2 100644 --- a/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py +++ b/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py @@ -196,10 +196,10 @@ def test_streaming_default_schema(self): "key": "a", "value": "b" }]), ("2", "bar", [0.3, 0.4], [{ "key": "c", "value": "d" - }]), ("3", "bar", [0.5, 0.6], [{ + }]), ("3", "foo", [0.5, 0.6], [{ "key": "e", "value": "f" }]), ("4", "bar", [0.7, 0.8], [{ - "key": "c", "value": "d" + "key": "g", "value": "h" }])]) ] args = self.test_pipeline.get_full_options_as_args(