add __all__ var to ConsAggShockModel
@@ -17,6 +17,10 @@ from HARK import HARKobject, Market, AgentType
from copy import deepcopy
# import matplotlib.pyplot as plt # All plotting is commented out
+__all__ = ['MargValueFunc2D', 'AggShockConsumerType', 'AggShockMarkovConsumerType',
+'CobbDouglasEconomy', 'SmallOpenEconomy', 'CobbDouglasMarkovEconomy',
+'SmallOpenMarkovEconomy', 'CobbDouglasAggVars', 'AggregateSavingRule', 'AggShocksDynamicRule']
+
utility = CRRAutility
utilityP = CRRAutilityP
utilityPP = CRRAutilityPP
@@ -25,7 +29,6 @@ utility_invP = CRRAutility_invP
utility_inv = CRRAutility_inv
-
class MargValueFunc2D(HARKobject):
'''
A class for representing a marginal value function in models where the
