From cfb30fe8c5646f408a645a445faea076d0b0386a Mon Sep 17 00:00:00 2001 From: Joris Van den Bossche Date: Tue, 11 Jun 2024 12:09:08 +0200 Subject: [PATCH] GH-41924: [Python] Fix tests when using NumPy 2.0 on Windows --- ci/appveyor-cpp-setup.bat | 2 ++ python/pyarrow/tests/parquet/common.py | 1 + 2 files changed, 3 insertions(+) diff --git a/ci/appveyor-cpp-setup.bat b/ci/appveyor-cpp-setup.bat index 5a9dffa166fb7..4690b6056d7a1 100644 --- a/ci/appveyor-cpp-setup.bat +++ b/ci/appveyor-cpp-setup.bat @@ -81,6 +81,8 @@ mamba create -n arrow -y -c conda-forge ^ "fsspec" ^ "python=%PYTHON%" ^ || exit /B +@rem TEMP test with numpy 2.0 RC +pip install --pre --upgrade numpy || exit /B conda list -n arrow @rem diff --git a/python/pyarrow/tests/parquet/common.py b/python/pyarrow/tests/parquet/common.py index 8365ed5b28543..53288729f9807 100644 --- a/python/pyarrow/tests/parquet/common.py +++ b/python/pyarrow/tests/parquet/common.py @@ -83,6 +83,7 @@ def _random_integers(size, dtype): iinfo = np.iinfo(dtype) return np.random.randint(max(iinfo.min, platform_int_info.min), min(iinfo.max, platform_int_info.max), + # size=size, dtype=dtype) size=size).astype(dtype)