Fix bigquery streaming test.

apache · Dec 20, 2024 · 2385897 · 2385897
1 parent 1ba9ee9
commit 2385897
Showing 1 changed file with 2 additions and 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
@@ -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(