scikit-learn · cmarmo · Aug 8, 2021 · Aug 10, 2021 · Aug 20, 2021 · Aug 20, 2021
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -159,6 +159,28 @@ jobs:
           paths:
             - ~/scikit_learn_data
 
+  linux-arm64-wheels:
+    machine:
+      image: ubuntu-2004:202101-01
+    resource_class: arm.medium
+    environment:
+      - CIBW_BUILD_VERBOSITY: 1
+      - CIBW_TEST_REQUIRES: pytest pandas threadpoolctl
+      - CIBW_TEST_COMMAND: bash {project}/build_tools/github/test_wheels.sh
+    steps:
+      - checkout
+      - run:
+          command: |
+            if [[ "$(git log --format=%B -n 1 $CIRCLE_SHA1)" =~ \[cd\ build\] ]]; then
+              bash build_tools/circle/checkout_merge_commit.sh
+              bash build_tools/circle/build_wheels_arm.sh
+            fi
+      - store_artifacts:
+          path: wheelhouse
+          destination: wheelhouse
+      - run: bash build_tools/github/upload_anaconda.sh
+
+
   deploy:
     docker:
       - image: circleci/python:3.7
@@ -177,7 +199,6 @@ jobs:
             fi
 
 workflows:
-  version: 2
   build-doc-and-deploy:
     jobs:
       - lint
@@ -208,3 +229,4 @@ workflows:
   linux-arm64:
     jobs:
       - linux-arm64
+      - linux-arm64-wheels
diff --git a/.travis.yml b/.travis.yml
diff --git a/README.rst b/README.rst
@@ -1,13 +1,10 @@
 .. -*- mode: rst -*-
 
-|Azure|_ |Travis|_ |Codecov|_ |CircleCI|_ |Nightly wheels|_ |Black|_ |PythonVersion|_ |PyPi|_ |DOI|_
+|Azure|_ |Codecov|_ |CircleCI|_ |Nightly wheels|_ |Black|_ |PythonVersion|_ |PyPi|_ |DOI|_
 
 .. |Azure| image:: https://dev.azure.com/scikit-learn/scikit-learn/_apis/build/status/scikit-learn.scikit-learn?branchName=main
 .. _Azure: https://dev.azure.com/scikit-learn/scikit-learn/_build/latest?definitionId=1&branchName=main
 
-.. |Travis| image:: https://api.travis-ci.com/scikit-learn/scikit-learn.svg?branch=main
-.. _Travis: https://travis-ci.com/scikit-learn/scikit-learn
-
 .. |Codecov| image:: https://codecov.io/gh/scikit-learn/scikit-learn/branch/main/graph/badge.svg?token=Pk8G9gg3y9
 .. _Codecov: https://codecov.io/gh/scikit-learn/scikit-learn
 

diff --git a/build_tools/circle/build_wheels_arm.sh b/build_tools/circle/build_wheels_arm.sh
@@ -0,0 +1,24 @@
+#!/bin/bash
+
+set -e
+set -x
+
+sudo apt-get update
+sudo apt-get install -y wget
+
+# Install Python 3.8 with conda from miniforge because this is
+# the simplest way to get a working anaconda-client for the wheel
+# upload step to the staging area on the anaconda.org service.
+export CONDA=$HOME/mambaforge
+wget https://github.com/conda-forge/miniforge/releases/latest/download/Mambaforge-$(uname)-$(uname -m).sh
+bash Mambaforge-$(uname)-$(uname -m).sh -b -f -p $CONDA
+echo 'export CONDA=$CONDA' >> $BASH_ENV  # to be reused in a later step
+export PATH=$CONDA/bin:$PATH
+mamba update --yes conda
+pip install --upgrade pip
+for pyversion in 37 38 39; do
+    pywheel=`echo "cp"$pyversion"-manylinux_aarch64"`
+    export CIBW_BUILD=$pywheel
+    python -m pip install cibuildwheel
+    python -m cibuildwheel --output-dir wheelhouse
+done
diff --git a/build_tools/github/check_wheels.py b/build_tools/github/check_wheels.py
@@ -17,18 +17,12 @@
 # 2 windows wheel (win32 + wind_amd64)
 n_wheels = 7 * n_python_versions
 
+# plus three more for linux/arm64 on circleci
+n_wheels += 3
+
 # plus one more for the sdist
 n_wheels += 1
 
-# aarch64 builds from travis
-travis_config_path = Path.cwd() / ".travis.yml"
-with travis_config_path.open("r") as f:
-    travis_config = yaml.safe_load(f)
-
-jobs = travis_config["jobs"]["include"]
-travis_builds = [j for j in jobs if any("CIBW_BUILD" in env for env in j["env"])]
-n_wheels += len(travis_builds)
-
 dist_files = list(Path("dist").glob("**/*"))
 n_dist_files = len(dist_files)
 

diff --git a/build_tools/github/upload_anaconda.sh b/build_tools/github/upload_anaconda.sh
@@ -3,6 +3,14 @@
 set -e
 set -x
 
+# Install Python 3.8 because of a bug with Python 3.9
+export PATH=$CONDA/bin:$PATH
+conda create -n upload -y python=3.8
+source activate upload
+conda install -y anaconda-client
+which anaconda
+
+# TODO: how to detect nightly builds for ARM on Circle CI?
 if [ "$GITHUB_EVENT_NAME" == "schedule" ]; then
     ANACONDA_ORG="scipy-wheels-nightly"
     ANACONDA_TOKEN="$SCIKIT_LEARN_NIGHTLY_UPLOAD_TOKEN"
@@ -11,12 +19,10 @@ else
     ANACONDA_TOKEN="$SCIKIT_LEARN_STAGING_UPLOAD_TOKEN"
 fi
 
-# Install Python 3.8 because of a bug with Python 3.9
-export PATH=$CONDA/bin:$PATH
-conda create -n upload -y python=3.8
-source activate upload
-conda install -y anaconda-client
-
-# Force a replacement if the remote file already exists
-anaconda -t $ANACONDA_TOKEN upload --force -u $ANACONDA_ORG dist/artifact/*
-echo "Index: https://pypi.anaconda.org/$ANACONDA_ORG/simple"
+if [ "$ANACONDA_TOKEN" == "" ]; then
+    echo "No upload token defined to upload to anaconda.org/$ANACONDA_ORG: skipping."
+else
+    # Force a replacement if the remote file already exists
+    anaconda -t $ANACONDA_TOKEN upload --force -u $ANACONDA_ORG dist/artifact/*
+    echo "Index: https://pypi.anaconda.org/$ANACONDA_ORG/simple"
+fi
diff --git a/build_tools/travis/after_success.sh b/build_tools/travis/after_success.sh
diff --git a/build_tools/travis/install.sh b/build_tools/travis/install.sh
diff --git a/build_tools/travis/install_main.sh b/build_tools/travis/install_main.sh
diff --git a/build_tools/travis/install_wheels.sh b/build_tools/travis/install_wheels.sh
diff --git a/build_tools/travis/script.sh b/build_tools/travis/script.sh
diff --git a/build_tools/travis/test_docs.sh b/build_tools/travis/test_docs.sh