importing scipy.sparse.linalg

tardis/montecarlo/montecarlo_numba/formal_integral.py

@@ -3,6 +3,7 @@
 import numpy as np
 import pandas as pd
 import scipy.sparse as sp
+import scipy.sparse.linalg as linalg
 from scipy.interpolate import interp1d
 from astropy import units as u
 from tardis import constants as const
@@ -414,7 +415,7 @@ def make_source_function(self):
                 inv_N = sp.identity(no_lvls) - Q
                 e_dot_u_vec = np.zeros(no_lvls)
                 e_dot_u_vec[e_dot_u_src_idx] = e_dot_u[shell].values
-                C_frame[shell] = sp.linalg.spsolve(inv_N.T, e_dot_u_vec)
+                C_frame[shell] = linalg.spsolve(inv_N.T, e_dot_u_vec)
 
         e_dot_u.index.names = [
             "atomic_number",