This is an automated email from the ASF dual-hosted git repository.
jrmccluskey 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 3c563e8d410 Fix postcommit python arm workflow (#33799)
3c563e8d410 is described below
commit 3c563e8d4103e84c8809b6379c7db9d703b6df2d
Author: akashorabek <[email protected]>
AuthorDate: Wed Jan 29 22:50:49 2025 +0500
Fix postcommit python arm workflow (#33799)
* Fix postcommit python arm workflow
* formatting
* formatting
---
sdks/python/apache_beam/ml/inference/sklearn_inference_it_test.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/sdks/python/apache_beam/ml/inference/sklearn_inference_it_test.py
b/sdks/python/apache_beam/ml/inference/sklearn_inference_it_test.py
index c5480234cda..c82b1698981 100644
--- a/sdks/python/apache_beam/ml/inference/sklearn_inference_it_test.py
+++ b/sdks/python/apache_beam/ml/inference/sklearn_inference_it_test.py
@@ -54,6 +54,9 @@ def file_lines_sorted(filepath):
@pytest.mark.uses_sklearn
@pytest.mark.it_postcommit
class SklearnInference(unittest.TestCase):
+
+ # TODO(https://github.com/apache/beam/issues/33796) use older numpy
+ @unittest.skipIf((3, 9, 0) <= sys.version_info < (3, 11, 0), "Beam#33796")
def test_sklearn_mnist_classification(self):
test_pipeline = TestPipeline(is_integration_test=True)
input_file = 'gs://apache-beam-ml/testing/inputs/it_mnist_data.csv'
@@ -85,6 +88,8 @@ class SklearnInference(unittest.TestCase):
true_label, expected_prediction = expected_outputs[i].split(',')
self.assertEqual(predictions_dict[true_label], expected_prediction)
+ # TODO(https://github.com/apache/beam/issues/33796) use older numpy
+ @unittest.skipIf((3, 9, 0) <= sys.version_info < (3, 11, 0), "Beam#33796")
def test_sklearn_mnist_classification_large_model(self):
test_pipeline = TestPipeline(is_integration_test=True)
input_file = 'gs://apache-beam-ml/testing/inputs/it_mnist_data.csv'
@@ -118,7 +123,8 @@ class SklearnInference(unittest.TestCase):
self.assertEqual(predictions_dict[true_label], expected_prediction)
# TODO(https://github.com/apache/beam/issues/27151) use model with sklearn
1.2
- @unittest.skipIf(sys.version_info >= (3, 11, 0), "Beam#27151")
+ # TODO(https://github.com/apache/beam/issues/33796) use older numpy
+ @unittest.skipIf(sys.version_info >= (3, 9, 0), "Beam#27151")
def test_sklearn_regression(self):
test_pipeline = TestPipeline(is_integration_test=True)
input_file =
'gs://apache-beam-ml/testing/inputs/japanese_housing_test_data.csv' # pylint:
disable=line-too-long
