Revision 0c53ce9def83699f107f3b5a3f683d0b2a50986a authored by George Necula on 02 April 2020, 09:13:40 UTC, committed by George Necula on 02 April 2020, 09:15:25 UTC
1 parent 1bb9aaa
.travis.yml
notifications:
email: false
dist: xenial
language: python
os: linux
jobs:
include:
- python: "3.6"
env: JAX_ONLY_CHECK_TYPES=true
- python: "3.6"
env: JAX_ENABLE_X64=0 JAX_NUM_GENERATED_CASES=25
- python: "3.6"
env: JAX_ENABLE_X64=1 JAX_NUM_GENERATED_CASES=25
- python: "3.7"
env: JAX_ENABLE_X64=1 JAX_ONLY_DOCUMENTATION=true
before_install:
- wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh;
- bash miniconda.sh -b -p $HOME/miniconda
- export PATH="$HOME/miniconda/bin:$PATH"
- conda config --set always_yes yes --set changeps1 no --set show_channel_urls yes
- conda config --set channel_priority strict
- conda config --set add_pip_as_python_dependency yes
- conda config --remove channels defaults
- conda config --add channels conda-forge
- conda update -q conda
install:
- conda install --yes python=$TRAVIS_PYTHON_VERSION pip absl-py opt_einsum numpy scipy pytest-xdist pytest-benchmark mypy
- if [ "$JAX_ONLY_CHECK_TYPES" = true ]; then
pip install pytype ;
fi
# The jaxlib version should match the minimum jaxlib version in
# jax/lib/__init__.py. This tests JAX PRs against the oldest permitted
# jaxlib.
- pip install jaxlib==0.1.43
- pip install -v .
# The following are needed to test the Colab notebooks and the documentation building
- if [ "$JAX_ONLY_DOCUMENTATION" = true ]; then
conda install --yes -c conda-forge pandoc ipykernel;
conda install --yes sphinx sphinx_rtd_theme nbsphinx sphinx-autodoc-typehints jupyter_client matplotlib;
pip install sklearn;
fi
script:
- if [ "$JAX_ONLY_DOCUMENTATION" = true ]; then
sphinx-build -b html -D nbsphinx_execute=always docs docs/build/html ;
elif [ "$JAX_ONLY_CHECK_TYPES" = true ]; then
echo "===== Checking with mypy ===="
time mypy --config-file=mypy.ini jax ;
else
pytest tests examples -W ignore ;
fi
Computing file changes ...