Skip to content

Commit b7b8bad

Browse files
committed
Fix experiments
1 parent 9024e26 commit b7b8bad

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -302,8 +302,7 @@ 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-
305+
pipeline_args = [arg for arg in pipeline_args if not arg.startswith("--experiments")]
307306

308307
pipeline_options = PipelineOptions(pipeline_args)
309308
pipeline = beam.Pipeline(options=pipeline_options)

0 commit comments

Comments
 (0)