12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- #!/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}
- 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
|