build: simplify test workflow

.github/workflows/tests.yml

@@ -27,17 +27,9 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install flake8 pytest mypy
+        python -m pip install pytest
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
-    - name: Lint with flake8
-      run: |
-        # stop the build if there are Python syntax errors or undefined names
-        flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
-        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
-        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
     - name: Test with pytest
       run: |
-        pytest
-    - name: Type hinting with MyPy
-      run: |
-        mypy bayex/
+        python -m pip install -e .
+        python -m pytest

requirements.txt

@@ -1,2 +1,3 @@
+jaxlib>=0.4.0
 jax
 numpy