Skip to content

Commit

Permalink
Run tests without JAX fixed random seed
Browse files Browse the repository at this point in the history
  • Loading branch information
agriyakhetarpal committed Mar 30, 2024
1 parent 3b39e9b commit dc7eda9
Showing 1 changed file with 44 additions and 0 deletions.
44 changes: 44 additions & 0 deletions .github/workflows/jax_reproducibility.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
name: JAX reproducibility report

on:
workflow_dispatch:
pull_request:

env:
FORCE_COLOR: 3

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
cancel-in-progress: true

jobs:
test:
name: Stress test / ${{ matrix.os }}
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-latest, macos-latest]
steps:
- uses: actions/checkout@v4
- name: Setup Python
uses: actions/setup-python@v5
with:
python-version: 3.12

- name: Install PyBaMM
run: |
pip install -e ."[all,dev,jax]"
- name: Run tests on ${{ matrix.os }}
shell: bash
run:
for i in {1..100}
do
python -m pytest tests/unit/test_solvers/test_jax_bdf_solver.py::TestJaxBDFSolver::test_solver_ >> logs-${{ matrix.os }}.txt || true
done

- name: Upload logs
uses: actions/upload-artifact@v4
with:
name: logs-${{ matrix.os }}-${{ github.run_id }}
path: logs-*.txt

0 comments on commit dc7eda9

Please sign in to comment.