.travis.yml

# Make it explicit that we favor the
# new container-based Travis workers
language: python
dist: xenial

cache:
  apt: true
  directories:
    - $HOME/.cache/pip
    - $HOME/.ccache

env:
  global:
    - CPU_COUNT=3
    - TEST_DIR=/tmp/sklearn  # Test directory for continuous integration jobs
    - PYTEST_VERSION=latest
    - OMP_NUM_THREADS=2
    - OPENBLAS_NUM_THREADS=2
    - SKLEARN_BUILD_PARALLEL=3
    - SKLEARN_SKIP_NETWORK_TESTS=1
    # Custom environment variables for the ARM wheel builder
    - CIBW_BUILD_VERBOSITY=1
    - CIBW_TEST_REQUIRES="pytest pytest-xdist threadpoolctl"
    - CIBW_TEST_COMMAND="bash {project}/build_tools/travis/test_wheels.sh"
    - CIBW_ENVIRONMENT="CPU_COUNT=2
                        OMP_NUM_THREADS=2
                        OPENBLAS_NUM_THREADS=2
                        SKLEARN_BUILD_PARALLEL=3
                        SKLEARN_SKIP_NETWORK_TESTS=1"

jobs:
  include:
    # Manual trigger of linux/arm64 tests in PR without triggering the full
    # wheel building process for all the Python versions.
    - python: 3.9
      os: linux
      arch: arm64
      if: commit_message =~ /\[arm64\]/
      env:
        - CPU_COUNT=4

    # Linux environments to build the scikit-learn wheels for the ARM64
    # architecture and Python 3.6 and newer. This is used both at release time
    # with the manual trigger in the commit message in the release branch and as
    # a scheduled task to build the weekly dev build on the main branch. The
    # weekly frequency is meant to avoid depleting the Travis CI credits too
    # fast.
    - python: 3.6
      os: linux
      arch: arm64
      if: type = cron or commit_message =~ /\[cd build\]/
      env:
        - BUILD_WHEEL=true
        - CIBW_BUILD=cp36-manylinux_aarch64

    - python: 3.7
      os: linux
      arch: arm64
      if: type = cron or commit_message =~ /\[cd build\]/
      env:
        - BUILD_WHEEL=true
        - CIBW_BUILD=cp37-manylinux_aarch64

    - python: 3.8
      os: linux
      arch: arm64
      if: type = cron or commit_message =~ /\[cd build\]/
      env:
        - BUILD_WHEEL=true
        - CIBW_BUILD=cp38-manylinux_aarch64

    - python: 3.9
      os: linux
      arch: arm64
      if: type = cron or commit_message =~ /\[cd build\]/
      env:
        - BUILD_WHEEL=true
        - CIBW_BUILD=cp39-manylinux_aarch64

install: source build_tools/travis/install.sh || travis_terminate 1
script: source build_tools/travis/script.sh || travis_terminate 1
after_success: source build_tools/travis/after_success.sh || travis_terminate 1

notifications:
  webhooks:
    urls:
      - https://webhooks.gitter.im/e/4ffabb4df010b70cd624
    on_success: change
    on_failure: always
    on_start: never