Adding test cases to composite mass modeling

hierarc/LensPosterior/kin_constraints_composite.py

@@ -216,21 +216,11 @@ def j_kin_draw_composite(self, kwargs_anisotropy, gamma_in, m2l, no_error=False)
 
         kwargs_lens_stars["amp"] *= m2l
 
-        if "sigma" in kwargs_lens_stars:
-            kwargs_lens_stars["sigma"] *= delta_r_eff
-        elif "Rs" in kwargs_lens_stars:
-            kwargs_lens_stars["Rs"] *= delta_r_eff
-        elif "R_sersic" in kwargs_lens_stars:
-            kwargs_lens_stars["R_sersic"] *= delta_r_eff
+        kwargs_lens_stars["sigma"] *= delta_r_eff
 
         kwargs_light = copy.deepcopy(self._kwargs_lens_light)
         for kwargs in kwargs_light:
-            if "sigma" in kwargs:
-                kwargs["sigma"] *= delta_r_eff
-            elif "Rs" in kwargs:
-                kwargs["Rs"] *= delta_r_eff
-            elif "R_sersic" in kwargs:
-                kwargs["R_sersic"] *= delta_r_eff
+            kwargs["sigma"] *= delta_r_eff
 
         kwargs_lens = [
             {

hierarc/Likelihood/parameter_scaling.py

@@ -29,15 +29,15 @@ def __init__(self, param_arrays, scaling_grid):
                 self._f_ani = interp2d(param_arrays[0], param_arrays[1], scaling_grid.T)
             else:
                 self._f_ani = RegularGridInterpolator(
-                    (arr for arr in param_arrays),
+                    tuple(param_arrays),
                     scaling_grid,
                 )
             self._evalute_scaling = True
 
     def param_scaling(self, param_array):
         """
 
-        :param aniso_param_array: anisotropy parameter array
+        :param param_array: anisotropy parameter array
         :return: scaling J(a_ani) for single slit
         """
         if self._evalute_scaling is not True or param_array is None:
@@ -59,10 +59,8 @@ def __init__(
         """
 
         :param anisotropy_model: string, either 'NONE', 'OM' or 'GOM'
-        :param param_array: array of parameter values
+        :param param_arrays: array of parameter values
         :param scaling_grid_list: list of array with the scalings of J() for each IFU
-        :param gamma_in_array: array of gamma_in values
-        :param m2l_array: array of M/L values
         """
         self._anisotropy_model = anisotropy_model
         self._evalute_ani = False