From 30d5ec2b58093764d795938977a4e868a5d19e88 Mon Sep 17 00:00:00 2001 From: Bhooshan Gadre Date: Mon, 24 Jul 2023 07:15:54 -0700 Subject: [PATCH] fixing relative import --- pycbc/waveform/gwsignal_utils.py | 9 ++++++--- pycbc/waveform/waveform.py | 2 +- pycbc/waveform/waveform_modes.py | 2 +- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/pycbc/waveform/gwsignal_utils.py b/pycbc/waveform/gwsignal_utils.py index c478b751649..8c91805f255 100644 --- a/pycbc/waveform/gwsignal_utils.py +++ b/pycbc/waveform/gwsignal_utils.py @@ -1,7 +1,7 @@ ''' Code to covert pycbc waveform gen params to gwsignal params ''' -import lalsimulation as lalsim +# import lalsimulation as lalsim from lalsimulation.gwsignal.core.parameter_conventions import (default_dict, common_units_dictionary) @@ -27,9 +27,12 @@ def to_gwsignal_dict(par): ''' params = par.copy() for key in par: + # if par[key]: knew = pycbc_to_gws.get(key, key) params[knew] = params.pop(key) - params[knew] *= gws_units.get(knew, 1.) - params.setdefault('condition', 1) + params[knew] = (params[knew]*gws_units.get(knew)) if params[knew] \ + else params[knew] + + _ = params.setdefault('condition', 1) return params diff --git a/pycbc/waveform/waveform.py b/pycbc/waveform/waveform.py index 7826677474f..84e3ff59a6b 100644 --- a/pycbc/waveform/waveform.py +++ b/pycbc/waveform/waveform.py @@ -44,7 +44,7 @@ zeros) from pycbc.waveform import parameters from pycbc.waveform import utils as wfutils -from .waveform.gwsignal_utils import to_gwsignal_dict +from .gwsignal_utils import to_gwsignal_dict from .spa_tmplt import (spa_amplitude_factor, spa_length_in_time, spa_tmplt, spa_tmplt_end, spa_tmplt_norm, spa_tmplt_precondition) diff --git a/pycbc/waveform/waveform_modes.py b/pycbc/waveform/waveform_modes.py index 2374aeeeb5d..2a0ef6219ce 100644 --- a/pycbc/waveform/waveform_modes.py +++ b/pycbc/waveform/waveform_modes.py @@ -24,7 +24,7 @@ from . import parameters from .waveform import _check_lal_pars, check_args, props -from .waveform.gwsignal_utils import to_gwsignal_dict +from .gwsignal_utils import to_gwsignal_dict from lalsimulation.gwsignal.core import waveform as wfm from lalsimulation.gwsignal.models import gwsignal_get_waveform_generator