Try reducing disk space in github actions

.github/workflows/run_pytest.yml

@@ -18,7 +18,7 @@ jobs:
       with:
         python-version: "3.8"
     - uses: actions/checkout@v3
-    - name: Install dependencies
+    - name: Install python dependencies
       run: |
         python -m pip install --upgrade pip
         pip install --upgrade pytest "jax[cpu]" flax torch tensorflow einops dask tinygrad scipy
@@ -35,12 +35,17 @@ jobs:
       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
+    - name: Set up OS
+      run: |
+        sudo rm -rf /usr/share/dotnet || true
+        sudo rm -rf /opt/ghc || true
+        sudo rm -rf /usr/local/.ghcup || true
+        sudo apt-get autoremove -y
+        sudo apt-get clean -y
+        sudo apt update -y
+        sudo apt install cmake liblapacke-dev libopenblas-dev liblapack-dev git -y
     - uses: actions/checkout@v3
-    - name: Install dependencies
+    - name: Install python dependencies
       run: |
         python -m pip install --upgrade pip
         pip install --upgrade pytest "jax[cpu]" flax dm-haiku torch tensorflow einops equinox dask tinygrad scipy