wehub-resource-sync
68 lines
2.0 KiB
Bash
Executable File
68 lines
2.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# This script automatically download ray and run the sanity check for Windows
|
|
# in various Python version. This script requires conda command to exist.
|
|
|
|
set -euo pipefail
|
|
|
|
export RAY_HASH="${RAY_HASH:-}"
|
|
export RAY_VERSION="${RAY_VERSION:-}"
|
|
|
|
if [[ -z "$RAY_HASH" ]]; then
|
|
echo "RAY_HASH env var should be provided"
|
|
exit 1
|
|
fi
|
|
|
|
if [[ -z "$RAY_VERSION" ]]; then
|
|
echo "RAY_VERSION env var should be provided"
|
|
exit 1
|
|
fi
|
|
|
|
if [[ ! -x "$(command -v conda)" ]]; then
|
|
echo "conda doesn't exist. Please download conda for this machine"
|
|
exit 1
|
|
else
|
|
echo "conda exists"
|
|
fi
|
|
|
|
echo "Start downloading Ray version ${RAY_VERSION} of commit ${RAY_HASH}"
|
|
|
|
python -m pip install --upgrade pip
|
|
|
|
# This is required to use conda activate
|
|
source "$(conda info --base)/etc/profile.d/conda.sh"
|
|
|
|
PYTHON_VERSIONS=( "3.10" "3.11" "3.12" )
|
|
|
|
for PYTHON_VERSION in "${PYTHON_VERSIONS[@]}"; do
|
|
ENV_NAME="${RAY_VERSION}-${PYTHON_VERSION}-env"
|
|
conda create -y -n "${ENV_NAME}" python="${PYTHON_VERSION}"
|
|
conda activate "${ENV_NAME}"
|
|
|
|
FAILED="false"
|
|
|
|
GOT_PYTHON_VERSION="$(python -c 'import sys; v=sys.version_info; print(f"{v.major}.{v.minor}")')"
|
|
if [[ "${GOT_PYTHON_VERSION}" != "${PYTHON_VERSION}" ]]; then
|
|
echo "Python version incorrect, got ${GOT_PYTHON_VERSION}, want ${PYTHON_VERSION}"
|
|
FAILED="true"
|
|
else
|
|
pip install --index-url 'https://test.pypi.org/simple/' --extra-index-url 'https://pypi.org/simple' "ray[cpp]==${RAY_VERSION}"
|
|
|
|
printf "\n\n\n"
|
|
echo "========================================================="
|
|
python sanity_check.py --ray_version="$RAY_VERSION" --ray_commit="$RAY_HASH"
|
|
echo "========================================================="
|
|
printf "\n\n\n"
|
|
fi
|
|
|
|
conda deactivate
|
|
conda remove -y --name "${ENV_NAME}" --all
|
|
|
|
if [[ "$FAILED" == "true" ]]; then
|
|
echo "PYTHON ${PYTHON_VERSION} failed sanity check."
|
|
exit 1
|
|
else
|
|
echo "PYTHON ${PYTHON_VERSION} succeed sanity check."
|
|
fi
|
|
done
|