Keyword for consumption (#9)

OpenSourceEconomics · Apr 24, 2024 · d0ec496 · d0ec496
1 parent c965e92
commit d0ec496
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/upper_envelope/fues_jax/fues_jax.py b/src/upper_envelope/fues_jax/fues_jax.py
@@ -797,7 +797,7 @@ def _compute_value(
     consumption, next_period_value, utility_function, utility_kwargs, discount_factor
 ):
     utility = utility_function(
-        consumption,
+        consumption=consumption,
         **utility_kwargs,
     )
     return utility + discount_factor * next_period_value
diff --git a/src/upper_envelope/fues_numba/fues_numba.py b/src/upper_envelope/fues_numba/fues_numba.py
@@ -782,7 +782,7 @@ def _augment_grids(
     )[:-1]
 
     utility = utility_function(
-        grid_points_to_add,
+        consumption=grid_points_to_add,
         **utility_kwargs,
     )
     values_to_add = utility + discount_factor * expected_value_zero_savings