Fix CI: apply ruff formatting and cleanup

Author: Math-llm-lab

notebooks/demo.ipynb

@@ -25,13 +25,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import math\n",
-    "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
+    "import numpy as np\n",
     "\n",
+    "from econ_math_portfolio.models.contract_stochastic_income import (\n",
+    "    ContractParams,\n",
+    "    lifetime_utility,\n",
+    "    solve_c_high,\n",
+    ")\n",
     "from econ_math_portfolio.models.cpi_target_discount import CpiParams, cpi, solve_t\n",
-    "from econ_math_portfolio.models.contract_stochastic_income import ContractParams, lifetime_utility, solve_c_high\n",
-    "from econ_math_portfolio.models.credit_var_quantile import CreditParams, var_analytic, var_mc\n"
+    "from econ_math_portfolio.models.credit_var_quantile import CreditParams, var_analytic, var_mc"
    ]
   },
   {
@@ -56,7 +59,7 @@
     "plt.ylabel(\"CPI(t)\")\n",
     "plt.show()\n",
     "\n",
-    "t_star\n"
+    "t_star"
    ]
   },
   {
@@ -81,7 +84,7 @@
     "plt.ylabel(\"V(c_high)\")\n",
     "plt.show()\n",
     "\n",
-    "c_star\n"
+    "c_star"
    ]
   },
   {
@@ -106,7 +109,7 @@
     "plt.ylabel(\"count\")\n",
     "plt.show()\n",
     "\n",
-    "analytic, mc_q\n"
+    "analytic, mc_q"
    ]
   }
  ],

src/econ_math_portfolio/__main__.py

@@ -1,4 +1,4 @@
 from .cli import main
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     raise SystemExit(main())

src/econ_math_portfolio/__pycache__/__init__.cpython-311.pyc

@@ -6,17 +6,26 @@
 from pathlib import Path
 from typing import Any
 
-from econ_math_portfolio.scoring import load_submission_json, score_submission, to_json_dict, load_rubric
+from econ_math_portfolio.scoring import (
+    load_rubric,
+    load_submission_json,
+    score_submission,
+    to_json_dict,
+)
+
 
 def _repo_root() -> Path:
     return Path(__file__).resolve().parents[2]
 
+
 def _validators_dir() -> Path:
     return _repo_root() / "validators"
 
+
 def _load_validator(task_id: str):
     return import_module(f"validators.{task_id}")
 
+
 def _emit(obj: Any, *, as_json: bool) -> None:
     if as_json:
         print(json.dumps(obj, indent=2, sort_keys=True))
@@ -26,23 +35,27 @@ def _emit(obj: Any, *, as_json: bool) -> None:
         else:
             print(obj)
 
+
 def cmd_list(*, as_json: bool) -> int:
     tasks = sorted([p.stem for p in _validators_dir().glob("*.py") if p.name != "__init__.py"])
     _emit({"tasks": tasks}, as_json=as_json)
     return 0
 
+
 def cmd_reference(task_id: str, *, as_json: bool) -> int:
     v = _load_validator(task_id)
     value = float(v.reference_compute())
     _emit({"task_id": task_id, "reference": value}, as_json=as_json)
     return 0
 
+
 def cmd_validate(task_id: str, answer: float, *, as_json: bool) -> int:
     v = _load_validator(task_id)
     res = v.validate(answer)
     _emit(res, as_json=as_json)
     return 0 if res["ok"] else 2
 
+
 def cmd_score(submission_path: str, *, as_json: bool) -> int:
     sub_path = Path(submission_path)
     payload = load_submission_json(sub_path)
@@ -67,8 +80,11 @@ def cmd_score(submission_path: str, *, as_json: bool) -> int:
     _emit(out, as_json=as_json)
     return 0 if sb.total >= 0.8 else 2
 
+
 def main(argv: list[str] | None = None) -> int:
-    p = argparse.ArgumentParser(prog="econ-math-portfolio", description="Math/Econ reasoning tasks + validators.")
+    p = argparse.ArgumentParser(
+        prog="econ-math-portfolio", description="Math/Econ reasoning tasks + validators."
+    )
     p.add_argument("--json", action="store_true", help="Output machine-readable JSON.")
     sub = p.add_subparsers(dest="cmd", required=True)
 

src/econ_math_portfolio/__pycache__/scoring.cpython-311.pyc

@@ -3,17 +3,22 @@
 import math
 from dataclasses import dataclass
 
+
 @dataclass(frozen=True)
 class ContractParams:
     delta: float = 0.95
     V0: float = 3.0
     c_low: float = 0.95
     autarky_high: float = 1.1
 
+
 def lifetime_utility(delta: float, c_low: float, c_high: float) -> float:
     return (0.5 * math.log(c_low) + 0.5 * math.log(c_high)) / (1.0 - delta)
 
-def solve_c_high(params: ContractParams, *, lo: float = 1e-8, hi: float = 10.0, iters: int = 200) -> float:
+
+def solve_c_high(
+    params: ContractParams, *, lo: float = 1e-8, hi: float = 10.0, iters: int = 200
+) -> float:
     def f(c_high: float) -> float:
         return lifetime_utility(params.delta, params.c_low, c_high) - params.V0