Update tests.yml

.github/workflows/tests.yml

@@ -24,6 +24,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install pytest
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+          if [ -f setup.py ]; then pip install .; fi
           pip install --upgrade "jax[cpu]"
       - name: Test with pytest
         run: |