Fix Python version check

apache · Sep 17, 2024 · d997350 · d997350
1 parent 7235140
commit d997350
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ci/scripts/python_wheel_unix_test.sh b/ci/scripts/python_wheel_unix_test.sh
@@ -57,6 +57,8 @@ export PYARROW_TEST_TENSORFLOW=ON
 export ARROW_TEST_DATA=${source_dir}/testing/data
 export PARQUET_TEST_DATA=${source_dir}/cpp/submodules/parquet-testing/data
 
+PYTHON_VERSION_MAJOR=$(python -c "import sys; print('.'.join(map(str, sys.version_info[:2])))")
+
 if [ "${INSTALL_PYARROW}" == "ON" ]; then
   # Install the built wheels
   pip install ${source_dir}/python/repaired_wheels/*.whl
@@ -96,7 +98,8 @@ fi
 if [ "${CHECK_UNITTESTS}" == "ON" ]; then
   # Install testing dependencies
   pip install -U -r ${source_dir}/python/requirements-wheel-test.txt
-  if [ ! "${PYTHON_VERSION}" = "3.13" ]; then
+
+  if [ ! "${PYTHON_VERSION_MAJOR}" = "3.13" ]; then
     pip install uwsgi
   fi