build-statsmodels.yml

# SPDX-FileCopyrightText: 2026 The RISE Project
# SPDX-License-Identifier: MIT
---
# Based on https://github.com/MacPython/statsmodels-wheels/blob/main/.github/workflows/build-wheels.yml
name: Build statsmodels wheels (riscv64)

on:
  workflow_dispatch:
    inputs:
      version:
        description: 'Version glob to (re)build; empty builds every version of docs/packages/statsmodels.yaml not released yet'
        required: false
        default: ''
  pull_request:
    branches: [main]
    paths:
      - '.github/workflows/build-statsmodels.yml'
      - 'docs/packages/statsmodels.yaml'
  push:
    branches: [main]
    paths:
      - '.github/workflows/build-statsmodels.yml'
      - 'docs/packages/statsmodels.yaml'

concurrency:
  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
  cancel-in-progress: true

permissions:
  contents: read  # to fetch code (actions/checkout)

env:
  MANYLINUX_RISCV64_IMAGE: quay.io/pypa/manylinux_2_39_riscv64

jobs:
  setup:
    uses: $/.github/workflows/_setup.yml
    with:
      package: statsmodels
      version: ${{ inputs.version }}

  build_wheels:
    needs: [setup]
    if: needs.setup.outputs.versions != '[]'
    name: Build statsmodels ${{ matrix.version }} ${{ matrix.python }}-manylinux_riscv64
    runs-on: ubuntu-24.04-riscv
    timeout-minutes: 720
    strategy:
      fail-fast: false
      matrix:
        version: ${{ fromJSON(needs.setup.outputs.versions) }}
        # scipy is a build requirement and pypi.riseproject.dev tops out at scipy
        # 1.15.2/cp313, so cp314 and cp314t cannot resolve one yet.
        python: ["cp312", "cp313"]

    env:
      STATSMODELS_VERSION: ${{ matrix.version }}

    steps:
      - name: Checkout statsmodels v${{ env.STATSMODELS_VERSION }}
        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
        with:
          repository: statsmodels/statsmodels
          ref: v${{ env.STATSMODELS_VERSION }}
          persist-credentials: false

      - name: Checkout python-wheels
        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1  # v7.0.1
        with:
          path: python-wheels
          persist-credentials: false

      - name: Patch statsmodels source
        run: git apply python-wheels/patches/statsmodels/${{ env.STATSMODELS_VERSION }}/*.patch

      - name: Build wheels
        uses: pypa/cibuildwheel@1828c10ab37f080699c7b81cea34097c684a7074  # v4.2.0
        with:
          output-dir: wheelhouse
          only: ${{ matrix.python }}-manylinux_riscv64
        env:
          CIBW_MANYLINUX_RISCV64_IMAGE: ${{ env.MANYLINUX_RISCV64_IMAGE }}
          # PIP_ONLY_BINARY keeps numpy/scipy/pandas on our registry's riscv64 wheels
          # instead of source-building the newer releases PyPI offers; setuptools_scm
          # is pinned because the patched tree is dirty at the tag. statsmodels'
          # meson.build calls setuptools_scm.get_version() directly with no dist_name,
          # so only the unscoped var (not _FOR_STATSMODELS) is honored.
          CIBW_ENVIRONMENT: >-
            SETUPTOOLS_SCM_PRETEND_VERSION=${{ env.STATSMODELS_VERSION }}
            PIP_EXTRA_INDEX_URL=https://pypi.riseproject.dev/simple/
            PIP_ONLY_BINARY=numpy,scipy,pandas
            MKL_NUM_THREADS=1
            OMP_NUM_THREADS=1
            OPENBLAS_NUM_THREADS=1
          CIBW_REPAIR_WHEEL_COMMAND_LINUX: auditwheel repair --strip -w {dest_dir} {wheel}
          # pytest 9.1 made strict_parametrization_ids the default, which rejects the
          # duplicate parametrize ids in statsmodels' test_moment_helpers at collection.
          CIBW_TEST_REQUIRES: 'pytest<9.1 pytest-xdist pytest-randomly threadpoolctl pytest-run-parallel'
          # test_seasonal_order[estimated] asserts a maximum-likelihood fit that
          # upstream itself logs as non-convergent lands within 1e-4 of the truth;
          # riscv64's BLAS stops it at 1.9e-3 where aarch64 reaches 2.4e-8.
          CIBW_TEST_COMMAND: >-
            python -c "import numpy; numpy.show_runtime(); import statsmodels;
            statsmodels.test(['-m', '(not slow and not example)', '-n', '2',
            '-k', 'not (test_seasonal_order and estimated)',
            '--randomly-dont-reset-seed'], exit=True)"

      - name: Check the wheel ships the compiled extensions
        run: |
          python3 - <<'EOF'
          import glob, zipfile

          names = zipfile.ZipFile(glob.glob("wheelhouse/*.whl")[0]).namelist()
          assert any(n.endswith(".so") for n in names), names
          assert any(n.endswith("LICENSE.txt") for n in names), names
          EOF

      - uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a  # v7.0.1
        with:
          name: statsmodels-${{ env.STATSMODELS_VERSION }}-${{ matrix.python }}-manylinux_riscv64
          path: wheelhouse/*.whl
          if-no-files-found: error

  publish:
    name: Publish statsmodels ${{ matrix.version }}
    needs: [setup, build_wheels]
    if: needs.setup.outputs.versions != '[]'
    strategy:
      fail-fast: false
      matrix:
        version: ${{ fromJSON(needs.setup.outputs.versions) }}
    permissions:
      contents: write
      pull-requests: write
    uses: $/.github/workflows/_publish-wheel.yml
    secrets:
      app-private-key: ${{ secrets.RISEPROJECT_APP_PRIVATE_KEY }}
    with:
      artifact-pattern: statsmodels-${{ matrix.version }}-*-manylinux_riscv64