line breaks for readability
@@ -181,7 +181,9 @@ class PrefShockConsumerType(IndShockConsumerType):
cNrmNow = np.zeros(self.AgentCount) + np.nan
for t in range(self.T_cycle):
these = t == self.t_cycle
- cNrmNow[these] = self.solution[t].cFunc(self.mNrmNow[these],self.shocks['PrefShkNow'][these])
+ cNrmNow[these] = self.solution[t].cFunc(
+ self.mNrmNow[these], self.shocks['PrefShkNow'][these]
+ )
self.cNrmNow = cNrmNow
return None
@@ -434,7 +436,10 @@ class ConsPrefShockSolver(ConsIndShockSolver):
for j in range(PrefShkCount):
this_shock = self.PrefShkVals[j]
this_prob = self.PrefShkPrbs[j]
- cNrmNow = solution.cFunc(mNrm_temp,this_shock*np.ones_like(mNrm_temp))
+ cNrmNow = solution.cFunc(mNrm_temp,
+ this_shock*np.ones_like(
+ mNrm_temp)
+ )
aNrmNow = mNrm_temp - cNrmNow
vNrmNow += this_prob*(this_shock*self.u(cNrmNow) + self.EndOfPrdvFunc(aNrmNow))
vPnow += this_prob*this_shock*self.uP(cNrmNow)
