fix calc_expectations bug
@@ -924,7 +924,6 @@ class ConsPortfolioIndShkRiskyAssetSolver(ConsIndShkRiskyAssetSolver):
EndOfPrddvds = calc_expectation(
self.RiskyDstn, endOfPrddvds, self.aNrmMat, self.shareMat
)
- EndOfPrddvds = EndOfPrddvds[:, :, 0]
self.risky_share_optimal = self.optimize_share(EndOfPrddvds)
@@ -1008,7 +1007,6 @@ class ConsPortfolioIndShkRiskyAssetSolver(ConsIndShkRiskyAssetSolver):
EndOfPrddvds = calc_expectation(
self.RiskyDstn, endOfPrddvds, self.aNrmMat, self.shareMat
)
- EndOfPrddvds = EndOfPrddvds[:, :, 0]
self.risky_share_optimal = self.optimize_share(EndOfPrddvds)
