Fixed return type hints

analysis/quadrupole_model_mappings.py

@@ -1,4 +1,4 @@
-from charged_shells import patch_size, parameters, charge_distributions
+from charged_shells import patch_size, parameters, charge_distributions, expansion
 from charged_shells import functions as fn
 from scipy.special import eval_legendre
 import numpy as np
@@ -19,7 +19,7 @@ def point_to_cap_magnitude(sigma_tilde: Array, a_bar: Array, theta0: Array, kapp
 
 
 def ic_to_gauss(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30,
-                sigma0: float = 0) -> charge_distributions.create_gaussian_charge_expansion:
+                sigma0: float = 0) -> expansion.Expansion:
 
     ex_mapped = charge_distributions.create_mapped_quad_expansion(a_bar=a_bar, kappaR=params.kappaR,
                                                                         sigma_tilde=sigma_tilde, l_max=30, sigma0=sigma0)
@@ -36,7 +36,7 @@ def ic_to_gauss(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30,
                                                                sigma1=gauss_sigma, l_max=l_max, sigma0=sigma0_mapped)
 
 
-def ic_to_cap(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30, sigma0: float = 0) -> charge_distributions.create_spherical_cap_expansion:
+def ic_to_cap(sigma_tilde, a_bar, params: ModelParams, l_max: int = 30, sigma0: float = 0) -> expansion.Expansion:
 
     ex_mapped = charge_distributions.create_mapped_quad_expansion(a_bar=a_bar, kappaR=params.kappaR,
                                                                         sigma_tilde=sigma_tilde, l_max=30, sigma0=sigma0)