From a310f5578a957a3a5b4e9a6ffb9a25d25355e710 Mon Sep 17 00:00:00 2001 From: jrmccluskey Date: Tue, 20 Oct 2026 16:11:08 +0000 Subject: [PATCH] Update test case for numpy 2.4.6 --- sdks/python/apache_beam/ml/transforms/base_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/ml/transforms/base_test.py b/sdks/python/apache_beam/ml/transforms/base_test.py index 64d0afe955d0..c7b904c09444 100644 --- a/sdks/python/apache_beam/ml/transforms/base_test.py +++ b/sdks/python/apache_beam/ml/transforms/base_test.py @@ -150,8 +150,8 @@ def test_ml_transform_on_list_dict(self): ), param( input_data=[{ - 'x': np.array([1], dtype=np.int64), - 'y': np.array([2.0], dtype=np.float32), + 'x': np.int64(1), + 'y': np.float32(2.0), }], input_types={ 'x': np.int32, 'y': np.float32