Check for discrete shares
@@ -1253,6 +1253,12 @@ def solveConsRiskyContrib(solution_next,ShockDstn,IncomeDstn,RiskyDstn,
# STEP THREE:
# Contribution share stage.
+ # TODO: implement continuous share case.
+ if not DiscreteShareBool:
+
+ raise Exception('The case of continuous shares has not been implemented yet')
+
+ else:
# Find the optimal share over the regular grid.
optIdx = np.argmax(vFxdNvrs, axis = 2)
@@ -1278,6 +1284,7 @@ def solveConsRiskyContrib(solution_next,ShockDstn,IncomeDstn,RiskyDstn,
# STEP FOUR:
# Rebalancing stage.
+ # Find optimal d for every combination
d = 'Here.' # TODO
