diff --git a/pycbc/waveform/waveform.py b/pycbc/waveform/waveform.py index 15cefeb87af..dd2476f4c78 100644 --- a/pycbc/waveform/waveform.py +++ b/pycbc/waveform/waveform.py @@ -1167,8 +1167,8 @@ def td_fd_waveform_transform(approximant): td_wav = _scheme.ChooseBySchemeDict() fd_wav = _scheme.ChooseBySchemeDict() -td_wav.update({_scheme.CPUScheme:cpu_td,_scheme.CUDAScheme:cuda_td,_scheme.CUPYScheme:cpu_td,}) -fd_wav.update({_scheme.CPUScheme:cpu_fd,_scheme.CUDAScheme:cuda_fd,_scheme.CUPYScheme:cpu_fd,}) +td_wav.update({_scheme.CPUScheme:cpu_td,_scheme.CUDAScheme:cuda_td}) +fd_wav.update({_scheme.CPUScheme:cpu_fd,_scheme.CUDAScheme:cuda_fd}) sgburst_wav = {_scheme.CPUScheme:cpu_sgburst} def get_waveform_filter(out, template=None, **kwargs):