Switch python3 with python

Some docker images have python3 set up as the system python with certain dependencies missing (e.g: auditwheel).

Author: nitins17

ci/build_artifacts.sh

@@ -20,22 +20,22 @@ source "ci/utilities/setup_build_environment.sh"
 
 # Build the jax artifact
 if [[ "$JAXCI_BUILD_JAX" == 1 ]]; then
-  python3 -m build --outdir $JAXCI_OUTPUT_DIR
+  python -m build --outdir $JAXCI_OUTPUT_DIR
 fi
 
 # Build the jaxlib CPU artifact
 if [[ "$JAXCI_BUILD_JAXLIB" == 1 ]]; then
-  python3 build/build.py jaxlib --ci_mode --request_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
+  python build/build.py jaxlib --ci_mode --request_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
 fi
 
 # Build the jax-cuda-plugin artifact
 if [[ "$JAXCI_BUILD_PLUGIN" == 1 ]]; then
-  python3 build/build.py jax-cuda-plugin --ci_mode --request_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
+  python build/build.py jax-cuda-plugin --ci_mode --request_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
 fi
 
 # Build the jax-cuda-pjrt artifact
 if [[ "$JAXCI_BUILD_PJRT" == 1 ]]; then
-  python3 build/build.py jax-cuda-pjrt --ci_mode --request_rbe --verbose
+  python build/build.py jax-cuda-pjrt --ci_mode --request_rbe --verbose
 fi
 
 # After building `jaxlib`, `jaxcuda-plugin`, and `jax-cuda-pjrt`, we run

ci/utilities/run_auditwheel.sh

@@ -28,7 +28,7 @@ fi
 for wheel in $WHEELS; do
     printf "\nRunning auditwheel on the following wheel:"
     ls $wheel
-    OUTPUT_FULL=$(python3 -m auditwheel show $wheel)
+    OUTPUT_FULL=$(python -m auditwheel show $wheel)
     # Remove the wheel name from the output to avoid false positives.
     wheel_name=$(basename $wheel)
     OUTPUT=${OUTPUT_FULL//${wheel_name}/}