[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/upper_envelope/upper_jor_drued.py

@@ -32,6 +32,7 @@ def upper_jor_drued(
 
     Returns arrays with the convention that index 0 corresponds to zero wealth:
     ``value_out[0] = expected_value_zero_savings`` and ``endog_out[0] = policy_out[0] = 0``.
+
     """
 
     if value_function_kwargs is None:

tests/test_upper_jor_drued.py

@@ -14,6 +14,7 @@
    non-degenerate reference segments (strictly increasing in `m`)
 3) interpolate the reference onto `m_grid` using only those safe segments
 4) compare `upper_jor_drued` to that reference on the masked points
+
 """
 
 from pathlib import Path
@@ -27,12 +28,13 @@
 
 import upper_envelope as upenv
 
-
 TEST_DIR = Path(__file__).parent
 TEST_RESOURCES_DIR = TEST_DIR / "resources"
 
 
-def utility_crra(consumption: jnp.ndarray, choice: int, params: Dict[str, float]) -> jnp.ndarray:
+def utility_crra(
+    consumption: jnp.ndarray, choice: int, params: Dict[str, float]
+) -> jnp.ndarray:
     utility_consumption = (consumption ** (1 - params["rho"]) - 1) / (1 - params["rho"])
     utility = utility_consumption - (1 - choice) * params["delta"]
     return utility
@@ -45,9 +47,10 @@ def interpolate_on_safe_reference_segments(
 ):
     """Interpolate reference (ref_m, ref_y) onto m_grid, ignoring unsafe segments.
 
-    A "safe" segment is any adjacent pair (ref_m[i], ref_m[i+1]) with ref_m[i+1] > ref_m[i].
-    For each x in m_grid, we take the maximum interpolated value over all safe segments
-    covering x. This avoids ambiguity around duplicated ref_m values.
+    A "safe" segment is any adjacent pair (ref_m[i], ref_m[i+1]) with ref_m[i+1] >
+    ref_m[i]. For each x in m_grid, we take the maximum interpolated value over all safe
+    segments covering x. This avoids ambiguity around duplicated ref_m values.
+
     """
 
     dm = ref_m[1:] - ref_m[:-1]
@@ -92,7 +95,9 @@ def test_upper_jor_drued_matches_fues_on_safe_segments(period, setup_model):
 
     def value_func(consumption, choice, params):
         # Same convention as existing tests: includes continuation value.
-        return utility_crra(consumption, choice, params) + params["beta"] * value_egm[1, 0]
+        return (
+            utility_crra(consumption, choice, params) + params["beta"] * value_egm[1, 0]
+        )
 
     ref_m, ref_c, ref_v = upenv.fues_jax(
         endog_grid=jnp.asarray(policy_egm[0, 1:]),