Try using awalsh128/cache-apt-pkgs-action@latest in run_pytest.yml

.github/workflows/run_pytest.yml

@@ -13,11 +13,11 @@ jobs:
   test_py38:
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v3
     - name: Set up python 3.8
       uses: actions/setup-python@v3
       with:
         python-version: "3.8"
+    - uses: actions/checkout@v3
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
@@ -31,14 +31,17 @@ jobs:
   test_py310:
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v3
     - name: Set up python 3.10
       uses: actions/setup-python@v3
       with:
         python-version: "3.10"
+    - uses: awalsh128/cache-apt-pkgs-action@latest
+      with:
+        packages: cmake liblapacke-dev libopenblas-dev liblapack-dev git
+        version: 1.0
+    - uses: actions/checkout@v3
     - name: Install dependencies
       run: |
-        apt update -y && apt install -y cmake liblapacke-dev libopenblas-dev liblapack-dev git
         python -m pip install --upgrade pip
         pip install --upgrade pytest "jax[cpu]" flax dm-haiku torch tensorflow einops equinox dask tinygrad scipy
         pip install --upgrade keras