Skip to content

Commit 2b43f80

Browse files
authored
Pin specifiable test to FnApiRunner (#36536)
Updated pipeline options to use 'FnApiRunner' and set pickle library.
1 parent 5d420c5 commit 2b43f80

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

sdks/python/apache_beam/yaml/yaml_specifiable_test.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,11 @@ def test_specifiable_transform(self):
5555
(0, beam.Row(x=4)),
5656
(0, beam.Row(x=9)),
5757
]
58-
with beam.Pipeline(options=beam.options.pipeline_options.PipelineOptions(
59-
pickle_library='cloudpickle')) as p:
58+
pipeline_options = beam.options.pipeline_options.PipelineOptions(
59+
pickle_library='cloudpickle')
60+
# Pin to FnApiRunner since this requires data from create to be
61+
# ppassed to anomaly detection in a certain order
62+
with beam.Pipeline('FnApiRunner', options=pipeline_options) as p:
6063
result = p | beam.Create(TRAIN_DATA) | YamlTransform(
6164
'''
6265
type: chain

0 commit comments

Comments
 (0)