Reference issue #30

OpenSourceEconomics · Feb 28, 2024 · 8769ad0 · 8769ad0
1 parent dcaf4a6
commit 8769ad0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/tests/test_models/phelps_deaton.py b/tests/test_models/phelps_deaton.py
@@ -29,7 +29,9 @@ def utility_with_filter(
     consumption,
     working,
     delta,
-    lagged_retirement,  # noqa: ARG001
+    # Temporary workaround for bug described in issue #30, which requires us to pass
+    # all state variables to the utility function.
+    lagged_retirement,  # noqa: ARG001, TODO: Remove unused arguments once #30 is fixed.
 ):
     return utility(consumption=consumption, working=working, delta=delta)
 

diff --git a/tests/test_models/stochastic.py b/tests/test_models/stochastic.py
@@ -20,7 +20,16 @@
 # --------------------------------------------------------------------------------------
 # Utility function
 # --------------------------------------------------------------------------------------
-def utility(consumption, working, health, partner, delta, gamma):  # noqa: ARG001
+def utility(
+    consumption,
+    working,
+    health,
+    # Temporary workaround for bug described in issue #30, which requires us to pass
+    # all state variables to the utility function.
+    partner,  # noqa: ARG001, TODO: Remove unused arguments once #30 is fixed.
+    delta,
+    gamma,
+):
     return jnp.log(consumption) + (gamma * health - delta) * working