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 4c5d9652bc5 Use timeout_ms
4c5d9652bc5 is described below
commit 4c5d9652bc555f125b27bacc600d5469215ad067
Author: Vitaly Terentyev <[email protected]>
AuthorDate: Tue Oct 6 16:40:40 2026 +0400
Use timeout_ms
---
...am_Inference_Python_Benchmarks_Dataflow_Pytorch_Image_Captioning.txt | 1 +
...Python_Benchmarks_Dataflow_Pytorch_Image_Classification_Rightfit.txt | 1 +
...erence_Python_Benchmarks_Dataflow_Pytorch_Image_Object_Detection.txt | 1 +
sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py | 2 +-
.../apache_beam/examples/inference/pytorch_image_object_detection.py | 2 +-
5 files changed, 5 insertions(+), 2 deletions(-)
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 c58707383ef..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
@@ -35,3 +35,4 @@
--clip_score_normalize=false
--runner=DataflowRunner
--experiments=use_runner_v2
+--timeout_ms=9000000
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 1140ee5735e..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
@@ -39,3 +39,4 @@
--inference_batch_size=16
--window_sec=60
--trigger_proc_time_sec=30
+--timeout_ms=9000000
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 18f017ad010..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
@@ -35,3 +35,4 @@
--model_state_dict_path=gs://apache-beam-ml/models/torchvision.detection.fasterrcnn_resnet50_fpn.pth
--runner=DataflowRunner
--experiments=use_runner_v2
+--timeout_ms=9000000
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 63165492b4c..f871bc362ea 100644
--- a/sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py
+++ b/sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py
@@ -618,7 +618,7 @@ def run(
method=method))
result = pipeline.run()
- result.wait_until_finish(duration=1800000) # 30 min
+ result.wait_until_finish(duration=9000000) # 150 min
result.cancel()
result.wait_until_finish(duration=600000) # up to 10 min to settle cancel
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 949fc48180f..eb870e3e95a 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
@@ -496,7 +496,7 @@ def run(
method=method))
result = pipeline.run()
- result.wait_until_finish(duration=1800000) # 30 min
+ result.wait_until_finish(duration=9000000) # 150 min
result.cancel()
result.wait_until_finish(duration=600000) # up to 10 min to settle cancel
