This is an automated email from the ASF dual-hosted git repository.
Amar3tto pushed a commit to branch fix-inference-ml
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/fix-inference-ml by this push:
new cacf15d01b2 Remove sdk_container_image and sdk_location
cacf15d01b2 is described below
commit cacf15d01b29703e8cc7c36f18cd24135e73f0bc
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Wed Oct 7 14:18:26 2026 +0400
Remove sdk_container_image and sdk_location
---
.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml | 2 +-
sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py | 3 ++-
.../apache_beam/examples/inference/pytorch_image_object_detection.py | 3 ++-
.../python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py | 3 ++-
4 files changed, 7 insertions(+), 4 deletions(-)
diff --git a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
index 4024fcd6c79..325e7efc100 100644
--- a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
+++ b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
@@ -111,7 +111,7 @@ jobs:
-Prunner=DataflowRunner \
-PpythonVersion=3.12 \
-PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_image_captioning_requirements.txt
\
- '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_13 }} --device=CPU
--mode=streaming
--job_name=benchmark-tests-pytorch-image-captioning-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_image_captioning_streaming_cpu
--output_table=apache-beam-testing.beam_run_inference.result_torch_inference_image_captioning_streaming_cpu'
+ '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_13 }} --device=CPU
--mode=streaming --sdk_location=container
--sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--job_name=benchmark-tests-pytorch-image-captioning-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_image_captioning_streaming_cpu
--output_table=apache-beam-testing.beam_run_inference.result_torch_inference_image_captioning_streaming_cpu'
- name: run PyTorch Image Object Detection Faster R-CNN ResNet-50
Streaming CPU
uses: ./.github/actions/gradle-command-self-hosted-action
timeout-minutes: 180
diff --git
a/sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py
b/sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py
index f871bc362ea..d6a8ddfce6c 100644
--- a/sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py
+++ b/sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py
@@ -497,7 +497,8 @@ def run_load_pipeline(known_args, pipeline_args):
pipeline_args, '--job_name', f"images-load-pubsub-{int(time.time())}")
override_or_add(pipeline_args, '--project', known_args.project)
pipeline_args = [
- arg for arg in pipeline_args if not arg.startswith("--experiments")
+ arg for arg in pipeline_args
+ if not arg.startswith(("--experiments", "--sdk_container_image",
"--worker_harness", "--sdk_location"))
]
pipeline_options = PipelineOptions(pipeline_args)
diff --git
a/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
b/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
index eb870e3e95a..614d85e12d7 100644
---
a/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
+++
b/sdks/python/apache_beam/examples/inference/pytorch_image_object_detection.py
@@ -378,7 +378,8 @@ def run_load_pipeline(known_args, pipeline_args):
pipeline_args, '--job_name', f"images-load-pubsub-{int(time.time())}")
override_or_add(pipeline_args, '--project', known_args.project)
pipeline_args = [
- arg for arg in pipeline_args if not arg.startswith("--experiments")
+ arg for arg in pipeline_args
+ if not arg.startswith(("--experiments", "--sdk_container_image",
"--worker_harness", "--sdk_location"))
]
pipeline_options = PipelineOptions(pipeline_args)
diff --git
a/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
b/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
index 1efb31fa240..bc3ab794107 100644
--- a/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
+++ b/sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py
@@ -386,7 +386,8 @@ def run_load_pipeline(known_args, pipeline_args):
pipeline_args, '--job_name', f"images-load-pubsub-{int(time.time())}")
override_or_add(pipeline_args, '--project', known_args.project)
pipeline_args = [
- arg for arg in pipeline_args if not arg.startswith("--experiments")
+ arg for arg in pipeline_args
+ if not arg.startswith(("--experiments", "--sdk_container_image",
"--worker_harness", "--sdk_location"))
]
pipeline_options = PipelineOptions(pipeline_args)
