a working waveform module for IMRPhenomParity

birefringence-waveform/__init__.py

-import .birefringence_waveform

birefringence-waveform/birefringence_waveform.py → waveform/IMRPhenomParity.py

+def genIMRPhenomParity(**kwds):
     from pycbc.waveform import get_fd_waveform
-#import numpy as np
-#import pycbc.conversions
     import lal
-import pycbc.waveform
-
-def IMRPhenomParity(parity_Amu, **kwds):
 
     if 'approximant' in kwds:
         kwds.pop("approximant")
     hp, hc = get_fd_waveform(approximant="IMRPhenomXPHM", **kwds)
 
-    temp =  parity_Amu / 1e9 / lal.QE_SI * lal.H_SI * lal.PI * lal.PI / lal.H0_SI
+    temp =  kwds['parity_Amu'] / 1e9 / lal.QE_SI * lal.H_SI * lal.PI * lal.PI / lal.H0_SI
 
     hp_parity = hp + hc * temp * hp.sample_frequencies **2
     hc_parity = hc - hp * temp * hp.sample_frequencies **2
@@ -18,4 +14,9 @@ def IMRPhenomParity(parity_Amu, **kwds):
     return hp_parity, hc_parity
 
 
-pycbc.waveform.add_custom_waveform('IMRPhenomParity', IMRPhenomParity, 'frequency', force=True)
+#pycbc.waveform.add_custom_waveform('IMRPhenomParity', IMRPhenomParity, 'frequency', force=True)
+
+def add_me(**kwds):
+    kwds['cpu_fd']['IMRPhenomParity'] = genIMRPhenomParity
+    kwds['filter_time_lengths']['IMRPhenomParity'] = kwds['filter_time_lengths']['IMRPhenomXPHM']
+

waveform/__init__.py

+from .IMRPhenomParity import *