You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by da...@apache.org on 2025/10/03 15:06:50 UTC
(beam) 01/01: Make ml tests portable
This is an automated email from the ASF dual-hosted git repository.
damccorm pushed a commit to branch users/damccorm/portableTest
in repository https://gitbox.apache.org/repos/asf/beam.git
commit 50738af972792bec76d5630bdbad09dd189a8c31
Author: Danny McCormick <da...@google.com>
AuthorDate: Fri Oct 3 11:06:38 2025 -0400
Make ml tests portable
---
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])