fix deps

.github/workflows/continuous-integration.yml

@@ -42,8 +42,8 @@ jobs:
         python -m pip install ipykernel
         python -m pip install grgrlib
         # for development versions of grgrjax
-        python -m pip uninstall -y grgrjax
-        python -m pip install git+https://github.com/gboehl/grgrjax
+        #python -m pip uninstall -y grgrjax
+        #python -m pip install git+https://github.com/gboehl/grgrjax
     - name: Test with pytest
       run: |
         pytest -x

econpizza/__version__.py

@@ -1,4 +1,4 @@
 #!/bin/python
 # -*- coding: utf-8 -*-
 
-__version__ = '0.5.3'
+__version__ = '0.5.4'

requirements.txt

@@ -1,6 +1,6 @@
 # ubuntu seems to have issues with jax v0.4.6. Recheck when v0.4.7 is out
 jax
 jaxlib
-grgrjax>=0.4.1
+grgrjax>=0.4.3
 scipy
 pyyaml

setup.py

@@ -34,9 +34,9 @@
         'linear': ['grgrlib>=0.1.22'],
     },
     install_requires=[
-        "jax",
-        "jaxlib",
-        "grgrjax>=0.4.1",
+        "jax<0.4.6",
+        "jaxlib<0.4.6",
+        "grgrjax>=0.4.3",
         "pyyaml",
         "scipy",
     ],