From 5d34d3c3a873d7e28421686304e623c63dd9b674 Mon Sep 17 00:00:00 2001 From: Wassim Kabalan Date: Sun, 8 Dec 2024 22:46:57 +0100 Subject: [PATCH] add test workflow --- .github/workflows/tests.yml | 41 +++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100644 .github/workflows/tests.yml diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml new file mode 100644 index 0000000..b716a49 --- /dev/null +++ b/.github/workflows/tests.yml @@ -0,0 +1,41 @@ +name: Tests + +on: + push: + branches: + - main + pull_request: + branches: + - main + +jobs: + build: + + runs-on: ubuntu-latest + strategy: + matrix: + python-version: ["3.10" , "3.11" , "3.12"] + + steps: + - name: Checkout Source + uses: actions/checkout@v2.3.1 + + - name: Set up Python ${{ matrix.python-version }} + uses: actions/setup-python@v2 + with: + python-version: ${{ matrix.python-version }} + + - name: Install dependencies + run: | + sudo apt-get install -y libopenmpi-dev + python -m pip install --upgrade pip + pip install jax=0.4.35 + pip install .[test] + + - name: Run Single Device Tests + run: | + cd tests + pytest -v -m "not distributed" + - name: Run Distributed tests + run: | + pytest -v -m distributed