update failing notebook
" \"mean\": init_parameters[\"PermGroFac\"],\n",
" \"std\": init_parameters[\"PermShkStd\"],\n",
" },\n",
- " ).approx(init_parameters[\"PermShkCount\"], tail_N=0),\n",
+ " ).discretize(\n",
+ " init_parameters[\"PermShkCount\"], method=\"equiprobable\", tail_N=0\n",
+ " ),\n",
" ),\n",
" Frame(\n",
" (\"TranShk\"),\n",
" }, # maybe this is unnecessary because the shock gets sampled at t = 0\n",
" transition=IndexDistribution(\n",
" MeanOneLogNormal, {\"sigma\": init_parameters[\"TranShkStd\"]}\n",
- " ).approx(init_parameters[\"TranShkCount\"], tail_N=0),\n",
+ " ).discretize(\n",
+ " init_parameters[\"TranShkCount\"], method=\"equiprobable\", tail_N=0\n",
+ " ),\n",
" ),\n",
" Frame(\n",
" (\"Rport\"),\n",
" \"mean\": init_parameters[\"PermGroFac\"],\n",
" \"std\": init_parameters[\"PermShkStd\"],\n",
" },\n",
- " ).approx(init_parameters[\"PermShkCount\"], tail_N=0),\n",
+ " ).discretize(\n",
+ " init_parameters[\"PermShkCount\"], method=\"equiprobable\", tail_N=0\n",
+ " ),\n",
" ),\n",
" Frame(\n",
" (\"TranShk\"),\n",
" }, # maybe this is unnecessary because the shock gets sampled at t = 0\n",
" transition=IndexDistribution(\n",
" MeanOneLogNormal, {\"sigma\": init_parameters[\"TranShkStd\"]}\n",
- " ).approx(init_parameters[\"TranShkCount\"], tail_N=0),\n",
+ " ).discretize(\n",
+ " init_parameters[\"TranShkCount\"], method=\"equiprobable\", tail_N=0\n",
+ " ),\n",
" ),\n",
" Frame( ## TODO: Handle Risky as an Aggregate value\n",
" (\"Risky\"),\n",
" \"std\": init_parameters[\"RiskyStd\"],\n",
" }\n",
" # seed=self.RNG.integers(0, 2 ** 31 - 1) : TODO: Seed logic\n",
- " ).approx(init_parameters[\"RiskyCount\"]),\n",
+ " ).discretize(init_parameters[\"RiskyCount\"], method=\"equiprobable\"),\n",
" aggregate=True,\n",
" ),\n",
" Frame(\n",
