mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
45 lines
1.2 KiB
Bash
Executable file
45 lines
1.2 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
# Python version can be specified as 3.7, 3.8, 3.9, etc..
|
|
if [ -z "$1" ]; then
|
|
PYTHON_VERSION=${PYTHON-3.7}
|
|
else
|
|
if [ "$1" = "3.6" ]; then
|
|
PYTHON_VERSION=${PYTHON-3.6}
|
|
elif [ "$1" = "3.7" ]; then
|
|
PYTHON_VERSION=${PYTHON-3.7}
|
|
elif [ "$1" = "3.8" ]; then
|
|
PYTHON_VERSION=${PYTHON-3.8}
|
|
elif [ "$1" = "3.9" ]; then
|
|
PYTHON_VERSION=${PYTHON-3.9}
|
|
elif [ "$1" = "3.10" ]; then
|
|
PYTHON_VERSION=${PYTHON-3.10}
|
|
else
|
|
echo "Unsupported Python version."
|
|
exit 1
|
|
fi
|
|
fi
|
|
echo "Python version is ${PYTHON_VERSION}"
|
|
|
|
|
|
ROOT_DIR=$(cd "$(dirname "$0")/$(dirname "$(test -L "$0" && readlink "$0" || echo "/")")" || exit; pwd)
|
|
WORKSPACE_DIR="${ROOT_DIR}/../.."
|
|
|
|
# Installs conda and python 3.7
|
|
MINIMAL_INSTALL=1 PYTHON=${PYTHON_VERSION} "${WORKSPACE_DIR}/ci/env/install-dependencies.sh"
|
|
|
|
# Re-install Ray wheels
|
|
rm -rf "${WORKSPACE_DIR}/python/ray/thirdparty_files"
|
|
rm -rf "${WORKSPACE_DIR}/python/ray/pickle5_files"
|
|
eval "${WORKSPACE_DIR}/ci/ci.sh build"
|
|
|
|
# Install test requirements
|
|
python -m pip install -U \
|
|
pytest==7.0.1 \
|
|
numpy
|
|
|
|
# Train requirements.
|
|
# TODO: make this dynamic
|
|
if [ "${TRAIN_MINIMAL_INSTALL-}" = 1 ]; then
|
|
python -m pip install -U "ray[tune]"
|
|
fi
|