(beam) branch users/damccorm/ri_test updated: Add another potential timeout

[da...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

damccorm pushed a commit to branch users/damccorm/ri_test
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/users/damccorm/ri_test by this push:
     new fc2e1bbfbf3 Add another potential timeout
fc2e1bbfbf3 is described below

commit fc2e1bbfbf38f3d7f9ef074517cd16ed8c36a2ac
Author: Danny McCormick <da...@google.com>
AuthorDate: Wed Aug 28 20:15:39 2024 +0100

    Add another potential timeout
---
 sdks/python/apache_beam/ml/inference/base_test.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/sdks/python/apache_beam/ml/inference/base_test.py b/sdks/python/apache_beam/ml/inference/base_test.py
index 00f6563d5f1..cbe12443b31 100644
--- a/sdks/python/apache_beam/ml/inference/base_test.py
+++ b/sdks/python/apache_beam/ml/inference/base_test.py
@@ -891,9 +891,9 @@ class RunInferenceBaseTest(unittest.TestCase):
       with TestPipeline() as pipeline:
         # Start with bad example which gets timed out.
         # Then provide plenty of time for GC to happen.
-        examples = [20] + [1] * 60
-        expected_good = [1] * 60
-        expected_bad = [20]
+        examples = [20] + [1] * 30 + [20]
+        expected_good = [1] * 30
+        expected_bad = [20, 20]
         pcoll = pipeline | 'start' >> beam.Create(examples)
         main, other = pcoll | base.RunInference(
             FakeSlowModelHandler(