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 87e366827b3 Fix sdk location
87e366827b3 is described below
commit 87e366827b3c8a7213bcc5987ba730859be5c217
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Wed Oct 7 13:05:48 2026 +0400
Fix sdk location
---
.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml | 8 ++++----
...erence_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt | 2 --
..._Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt | 2 --
..._Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt | 2 --
4 files changed, 4 insertions(+), 10 deletions(-)
diff --git a/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
b/.github/workflows/beam_Inference_Python_Benchmarks_Dataflow.yml
index 95774a7fa6d..20b5fad3859 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
--job_name=benchmark-tests-pytorch-image-captioning-streaming-${{env.NOW_UTC}}
--sdk_location=container
--sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--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
@@ -122,7 +122,7 @@ jobs:
-Prunner=DataflowRunner \
-PpythonVersion=3.12 \
-PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_image_object_detection_requirements.txt
\
- '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_12 }} --device=CPU
--mode=streaming
--job_name=benchmark-tests-pytorch-image-object-detection-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_image_object_detection_streaming_cpu
--output_table=apache-beam-testing.beam_run_inference.result_torch_inference_image_object_detection_streaming_cpu'
\
+ '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_12 }} --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-object-detection-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_image_object_detection_streaming_cpu
--output_table=apache-beam-testing.beam_run_inference.result_torch_inference_image_object_det
[...]
- name: run PyTorch Image Object Detection Faster R-CNN ResNet-50
Streaming GPU
uses: ./.github/actions/gradle-command-self-hosted-action
timeout-minutes: 180
@@ -155,7 +155,7 @@ jobs:
-Prunner=DataflowRunner \
-PpythonVersion=3.12 \
-PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_rightfit_requirements.txt
\
- '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_11 }} --device=CPU
--experiments=enable_streaming_rightfitting --mode=streaming
--job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_imagenet_stream_rightfit_cpu
--output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_cpu'
\
+ '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_11 }} --device=CPU
--sdk_location=container
--sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--experiments=enable_streaming_rightfitting --mode=streaming
--job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_imagenet_stream_rightfit_cpu
--output_table=apache-beam-testing.beam_run_inference.result_t [...]
- name: run PyTorch Image Classification EfficientNet-B0 Streaming
(Right-fitting Exactly-once) CPU
uses: ./.github/actions/gradle-command-self-hosted-action
timeout-minutes: 180
@@ -166,7 +166,7 @@ jobs:
-Prunner=DataflowRunner \
-PpythonVersion=3.12 \
-PloadTest.requirementsTxtFile=apache_beam/ml/inference/pytorch_rightfit_requirements.txt
\
- '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_11 }} --device=CPU
--experiments=enable_streaming_rightfitting --mode=streaming
--job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_imagenet_stream_rightfit_once_cpu
--output_table=apache-beam-testing.beam_run_inference.result_torch_inference_imagenet_stream_rightfit_once_cpu'
\
+ '-PloadTest.args=${{
env.beam_Inference_Python_Benchmarks_Dataflow_test_arguments_11 }} --device=CPU
--sdk_location=container
--sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--experiments=enable_streaming_rightfitting --mode=streaming
--job_name=benchmark-tests-pytorch-imagenet-rightfit-streaming-${{env.NOW_UTC}}
--metrics_table=torch_inference_imagenet_stream_rightfit_once_cpu
--output_table=apache-beam-testing.beam_run_inference.res [...]
- name: run PyTorch Image Classification EfficientNet-B0 Streaming
(Right-fitting) GPU
uses: ./.github/actions/gradle-command-self-hosted-action
timeout-minutes: 180
diff --git
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt
index d9b5ac250e0..92a731ccc7b 100644
---
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt
+++
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt
@@ -36,5 +36,3 @@
--runner=DataflowRunner
--experiments=use_runner_v2
--timeout_ms=9000000
---sdk_location=container
---sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
diff --git
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt
index 850838554a2..616d0d7c135 100644
---
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt
+++
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt
@@ -21,8 +21,6 @@
--autoscaling_algorithm=NONE
--staging_location=gs://temp-storage-for-perf-tests/loadtests
--temp_location=gs://temp-storage-for-perf-tests/loadtests
---sdk_location=container
---sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
--requirements_file=apache_beam/ml/inference/pytorch_rightfit_requirements.txt
--publish_to_big_query=true
--metrics_dataset=beam_run_inference
diff --git
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt
index 04a9618b1fd..98d00208409 100644
---
a/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt
+++
b/.github/workflows/load-tests-pipeline-options/beam_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt
@@ -36,5 +36,3 @@
--runner=DataflowRunner
--experiments=use_runner_v2
--timeout_ms=9000000
---sdk_location=container
---sdk_container_image=us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest
