Skip to content

Commit 9cd994e

Browse files
authored
Merge pull request #36302 from tarun-google/fix_bigquery_tests
Fix BigQuery Batch write Integration tests
2 parents bacd18b + 08e76a0 commit 9cd994e

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

sdks/python/apache_beam/yaml/extended_tests/data/enrichment.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,9 @@ pipelines:
4040
- type: WriteToBigQuery
4141
config:
4242
table: "{BQ_TABLE}"
43+
options:
44+
project: "apache-beam-testing"
45+
temp_location: "{TEMP_DIR}"
4346

4447
- pipeline:
4548
type: chain

sdks/python/apache_beam/yaml/integration_tests.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -721,8 +721,9 @@ def test(self, providers=providers): # default arg to capture loop value
721721
for pipeline_spec in spec['pipelines']:
722722
with beam.Pipeline(options=PipelineOptions(
723723
pickle_library='cloudpickle',
724-
**yaml_transform.SafeLineLoader.strip_metadata(pipeline_spec.get(
725-
'options', {})))) as p:
724+
**replace_recursive(yaml_transform.SafeLineLoader.strip_metadata(
725+
pipeline_spec.get('options', {})),
726+
vars))) as p:
726727
yaml_transform.expand_pipeline(
727728
p, replace_recursive(pipeline_spec, vars))
728729

0 commit comments

Comments
 (0)