Skip to content

Commit 9024e26

Browse files
committed
Fix experiments
1 parent 12cd477 commit 9024e26

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -302,6 +302,8 @@ def run_load_pipeline(known_args, pipeline_args):
302302
override_or_add(pipeline_args, '--max_num_workers', '10')
303303
override_or_add(pipeline_args, '--job_name', f"images-load-pubsub-{int(time.time())}")
304304
override_or_add(pipeline_args, '--project', known_args.project)
305+
override_or_add(pipeline_args, '--experiments', 'use_runner_v2')
306+
305307

306308
pipeline_options = PipelineOptions(pipeline_args)
307309
pipeline = beam.Pipeline(options=pipeline_options)

0 commit comments

Comments
 (0)