diff --git a/tupak/gw/source.py b/tupak/gw/source.py index 81ab36beb079f8c0ae88939e1f916e0983d2a277..afe9d0cd0f21cdeeea152c88b613c85773815c88 100644 --- a/tupak/gw/source.py +++ b/tupak/gw/source.py @@ -146,10 +146,13 @@ def supernova( def supernova_pca_model( - frequency_array, realPCs, imagPCs, pc_coeff1, pc_coeff2, pc_coeff3, - pc_coeff4, pc_coeff5, luminosity_distance, ra, dec, geocent_time, psi): + frequency_array, pc_coeff1, pc_coeff2, pc_coeff3, pc_coeff4, pc_coeff5, + luminosity_distance, ra, dec, geocent_time, psi, **kwargs): """ Supernova signal model """ + realPCs = kwargs['realPCs'] + imagPCs = kwargs['imagPCs'] + pc1 = realPCs[:, 0] + 1.0j*imagPCs[:, 0] pc2 = realPCs[:, 1] + 1.0j*imagPCs[:, 1] pc3 = realPCs[:, 2] + 1.0j*imagPCs[:, 2]