From d4c04174b41da5e5b7a201c512792600097bee87 Mon Sep 17 00:00:00 2001 From: Akarys Shorabek Date: Wed, 29 Jan 2025 21:49:44 +0500 Subject: [PATCH] formatting --- .../apache_beam/ml/inference/sklearn_inference_it_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 179aafbf75b1..c82b1698981b 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 @@ -55,7 +55,7 @@ def file_lines_sorted(filepath): @pytest.mark.it_postcommit class SklearnInference(unittest.TestCase): - # TODO(https://github.com/apache/beam/issues/33796) use older numpy version for python 3.9 and 3.10 + # 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) @@ -88,7 +88,7 @@ def test_sklearn_mnist_classification(self): 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 version for python 3.9 and 3.10 + # 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) @@ -123,7 +123,7 @@ def test_sklearn_mnist_classification_large_model(self): self.assertEqual(predictions_dict[true_label], expected_prediction) # TODO(https://github.com/apache/beam/issues/27151) use model with sklearn 1.2 - # TODO(https://github.com/apache/beam/issues/33796) use older numpy version for python 3.9 and 3.10 + # 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)