Skip to content

run levanter tests in Haliax #1

run levanter tests in Haliax

run levanter tests in Haliax #1

name: Run Tests
on: [push, pull_request]
jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- name: Set up Python 3.10.11
uses: actions/setup-python@v4
with:
python-version: 3.10.11
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install flake8 pytest
pip install --upgrade "jax[cpu]==0.4.25" "jaxlib[cpu]==0.4.25"
- name: Install Levanter from source
run: |
git clone "git+https://github.com/stanford-crfm/levanter.git"
cd levanter
pip install .
- name: Install Haliax on top
run: |
# install levanter main
cd ..
pip install .[dev]
- name: Test levanter with pytest
run: |
cd levanter
XLA_FLAGS=--xla_force_host_platform_device_count=8 PYTHONPATH=tests:src:../src pytest tests -m "not entry and not slow"