(beam) branch master updated: Make ml tests portable (#36377)

[damccorm]

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

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


The following commit(s) were added to refs/heads/master by this push:
     new bd5cfdacc07 Make ml tests portable (#36377)
bd5cfdacc07 is described below

commit bd5cfdacc07018628b2871ff488b9a772b88ab02
Author: Danny McCormick <dannymccorm...@google.com>
AuthorDate: Fri Oct 3 13:13:35 2025 -0400

    Make ml tests portable (#36377)
---
 sdks/python/apache_beam/ml/inference/base_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sdks/python/apache_beam/ml/inference/base_test.py 
b/sdks/python/apache_beam/ml/inference/base_test.py
index 64fd73682e1..66e85ce163e 100644
--- a/sdks/python/apache_beam/ml/inference/base_test.py
+++ b/sdks/python/apache_beam/ml/inference/base_test.py
@@ -1141,7 +1141,7 @@ class RunInferenceBaseTest(unittest.TestCase):
             accumulation_mode=trigger.AccumulationMode.DISCARDING))
 
     test_pipeline.options.view_as(StandardOptions).streaming = True
-    with self.assertRaises(ValueError) as e:
+    with self.assertRaises(Exception) as e:
       _ = (
           test_pipeline
           | beam.Create([1, 2, 3, 4])
@@ -1165,7 +1165,7 @@ class RunInferenceBaseTest(unittest.TestCase):
             accumulation_mode=trigger.AccumulationMode.DISCARDING))
 
     test_pipeline.options.view_as(StandardOptions).streaming = True
-    with self.assertRaises(ValueError) as e:
+    with self.assertRaises(Exception) as e:
       _ = (
           test_pipeline
           | beam.Create([1, 2, 3, 4])