hessboost 0.1.0

A faithful, fast, pure-Rust reimplementation of XGBoost gradient boosting
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143

name: CI

on:
  push:
    branches: [main]
  pull_request:
  workflow_dispatch:

concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true

permissions:
  contents: read

env:
  CARGO_INCREMENTAL: 0
  CARGO_TERM_COLOR: always
  RUSTFLAGS: "-D warnings"

# Toolchain and tool versions (rust, mbx, uv) come from mise.toml. mise-action
# runs before mr-boxington-action so the cache key sees the pinned rustc, and
# the cache action reuses the mbx that mise put on PATH. mise's own cache is
# off: it would restore the rust install symlink without rustup's toolchain.

jobs:
  fmt:
    name: rustfmt
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v7
        with:
          persist-credentials: false
      - uses: jdx/mise-action@v4
        with:
          cache: false
      - run: cargo fmt --all --check

  clippy:
    name: clippy
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v7
        with:
          persist-credentials: false
      - uses: jdx/mise-action@v4
        with:
          cache: false
      - uses: jdx/mr-boxington-action@v1
        with:
          cache-generation: clippy
      # Lint levels (pedantic) live in Cargo.toml's [lints] table.
      - run: mbx clippy --all-targets --all-features -- -D warnings

  test:
    name: test (${{ matrix.name }})
    runs-on: ${{ matrix.runner }}
    strategy:
      fail-fast: false
      matrix:
        include:
          - { name: x86_64-linux, runner: ubuntu-26.04 }
          - { name: aarch64-linux, runner: ubuntu-26.04-arm }
          - { name: aarch64-darwin, runner: macos-latest }
    steps:
      - uses: actions/checkout@v7
        with:
          persist-credentials: false
      - uses: jdx/mise-action@v4
        with:
          cache: false
      - uses: jdx/mr-boxington-action@v1
        with:
          cache-generation: test
      - run: mbx test --all-features

  docs:
    name: docs
    runs-on: ubuntu-latest
    env:
      RUSTDOCFLAGS: "-D warnings"
    steps:
      - uses: actions/checkout@v7
        with:
          persist-credentials: false
      - uses: jdx/mise-action@v4
        with:
          cache: false
      - uses: jdx/mr-boxington-action@v1
        with:
          cache-generation: docs
      - run: mbx doc --no-deps --all-features

  msrv:
    name: MSRV (1.93.0)
    runs-on: ubuntu-latest
    env:
      # Overrides the rust version from mise.toml for every mise step.
      MISE_RUST_VERSION: 1.93.0
    steps:
      - uses: actions/checkout@v7
        with:
          persist-credentials: false
      - uses: jdx/mise-action@v4
        with:
          cache: false
      - uses: jdx/mr-boxington-action@v1
        with:
          cache-generation: msrv
      - run: mbx build --all-features

  parity:
    name: XGBoost parity
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v7
        with:
          persist-credentials: false
      - uses: jdx/mise-action@v4
        with:
          cache: false
      - uses: jdx/mr-boxington-action@v1
        with:
          cache-generation: parity
      - name: Generate fixtures from XGBoost 3.4.1
        run: uv run --with xgboost==3.4.1 --with numpy python scripts/gen_fixtures.py
      - name: Train, import and cut parity
        run: mbx test --test parity --release -- --ignored --nocapture
      - name: Export parity (reload in XGBoost)
        run: uv run --with xgboost==3.4.1 --with numpy python scripts/check_exports.py

  all-checks-passed:
    permissions: {}
    needs: [fmt, clippy, test, docs, msrv, parity]
    runs-on: ubuntu-latest
    timeout-minutes: 1
    # Run on upstream failure too (so branch protection sees a failure), but
    # not on cancellation.
    if: ${{ !cancelled() }}
    steps:
      - if: contains(needs.*.result, 'failure') || contains(needs.*.result, 'cancelled') || contains(needs.*.result, 'skipped')
        run: exit 1
      - run: echo "All checks passed"