From 2b7e3fcbd7c1c52eb9a46bbde4172dc471280a5e Mon Sep 17 00:00:00 2001
From: Alon Kukliansky <alonkukl@gmail.com>
Date: Sat, 14 Sep 2024 22:03:48 -0700
Subject: [PATCH] Removing python3.8, and also making sure to only test CPU

---
 .github/workflows/tests.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index e9f6060..5e27489 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -10,7 +10,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ['3.8', '3.9', '3.10', '3.11', '3.12']
+        python-version: ['3.9', '3.10', '3.11', '3.12']
         os: [ubuntu-latest]
 
     runs-on: ${{ matrix.os }}
@@ -30,4 +30,5 @@ jobs:
     - name: Run tests
       env:
         AMOUNT_OF_WORKERS: 2
+        JAX_PLATFORMS: cpu
       run: pytest