FIX: Fix ModuleNotFoundError for Numba 0.49.0

QuantEcon · Apr 20, 2020 · 8dc5968 · 8dc5968
1 parent c4ade35
commit 8dc5968
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/quantecon/util/numba.py b/quantecon/util/numba.py
@@ -4,7 +4,10 @@
 """
 import numpy as np
 from numba import jit, generated_jit, types
-from numba.targets.linalg import _LAPACK
+try:
+ from numba.np.linalg import _LAPACK # for Numba >= 0.49.0
+except ModuleNotFoundError:
+ from numba.targets.linalg import _LAPACK # for Numba < 0.49.0
 
 
 # BLAS kinds as letters