FIX: Import scipy.sparse.linalg

Fix #481
QuantEcon · Apr 12, 2019 · ac3b6ab · ac3b6ab
1 parent 3029221
commit ac3b6ab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/quantecon/markov/ddp.py b/quantecon/markov/ddp.py
@@ -417,7 +417,8 @@ def s_wise_max(vals, out=None, out_argmax=None):
 
         # Linear equation solver to be used in evaluate_policy
         if self._sparse:
-            self._lineq_solve = sp.linalg.spsolve
+            import scipy.sparse.linalg
+            self._lineq_solve = scipy.sparse.linalg.spsolve
             self._I = sp.identity(self.num_states, format='csr')
         else:
             self._lineq_solve = np.linalg.solve