add numpy import

econ-ark · Sep 18, 2017 · 42f7d9f · 42f7d9f
1 parent 1d9fd69
commit 42f7d9f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ConsumptionSaving/TractableBufferStockModel.py b/ConsumptionSaving/TractableBufferStockModel.py
@@ -22,6 +22,7 @@
 # contained in the HARK folder. 
 import sys
 import os
+import numpy as np
 sys.path.insert(0, os.path.abspath('../'))
 
 from HARKcore import AgentType, NullFunc, Solution
@@ -557,4 +558,4 @@ def getPostStates(self):
     diffFunc = lambda m : ExampleType.solution[0].cFunc(m) - MarkovType.cFunc[0][0](m)
     print('Difference between the (employed) consumption functions:')
     plotFuncs(diffFunc,0,m_upper)
-
+