fix grgrjax dep + fix tests

Author: gboehl

econpizza/examples/hank_with_comments.yml

@@ -131,10 +131,10 @@ steady_state:
         B: 5.6 # bond supply
         # definitions can be recursive: theta is defined above
         mc: (theta-1)/theta # marginal costs
-        #w: mc*z # wages
-        #n: w**(1/sigma_l) # labor supply
-        #div: 1 - w*n # dividends
-        #z: y/n # technology
+        w: (mc*y)**(sigma_l/(1+sigma_l)) # wages
+        n: w**(1/sigma_l) # labor supply
+        z: y/n # technology
+        div: 1 - w*n # dividends
 
     init_guesses:
         Rstar: 1.002 # steady state target rage

econpizza/testing/cache/hank_labor.npy

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

econpizza/testing/cache/hank_solid.npy

@@ -36,7 +36,7 @@
     install_requires=[
         "jax>=0.4.13",
         "jaxlib>=0.4.13",
-        "grgrjax>=0.4.3",
+        "grgrjax>=0.4.4",
         "pyyaml",
         "scipy",
     ],