From 407e36f9f75722c2bba0fdf9bb918edbb5801559 Mon Sep 17 00:00:00 2001 From: Yueqi Wang Date: Thu, 26 Feb 2026 12:01:51 -0500 Subject: [PATCH] add CI yaml config for running tests to in PR checks --- .github/workflows/ci.yml | 38 ++++++++++++++++++++++++++++++++++++++ tests/__init__.py | 0 2 files changed, 38 insertions(+) create mode 100644 .github/workflows/ci.yml create mode 100644 tests/__init__.py diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml new file mode 100644 index 0000000..f672f11 --- /dev/null +++ b/.github/workflows/ci.yml @@ -0,0 +1,38 @@ +name: CI + +on: + pull_request: + branches: [main] + +jobs: + test: + runs-on: ubuntu-latest + + steps: + - uses: actions/checkout@v4 + + - name: Set up Python + uses: actions/setup-python@v5 + with: + python-version: "3.10" + + - name: Install PyTorch (CPU) + run: pip install torch --index-url https://download.pytorch.org/whl/cpu + + - name: Install JAX (CPU) + run: pip install "jax[cpu]" + + - name: Install package + run: pip install -e . + + - name: Run CSR builder tests + run: python -m tests.test_csr_builder + + - name: Run PyTorch decoding tests + run: python -m tests.test_pt_decoding + + - name: Run JAX decoding tests + run: python -m tests.test_jax_decoding + + - name: Run JAX baseline tests + run: python -m tests.test_baselines_jax diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29