Skip to content

Commit

Permalink
fix error when specifying orbits in space_detector
Browse files Browse the repository at this point in the history
  • Loading branch information
acorreia61201 committed Nov 18, 2024
1 parent 0eb2e7d commit 8b40470
Showing 1 changed file with 13 additions and 14 deletions.
27 changes: 13 additions & 14 deletions pycbc/detector/space.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,8 +255,8 @@ class _LDC_detector(AbsSpaceDet):
and TDI. See self.orbits_init for accepted inputs. Default
'EqualArmlength'.
"""
def __init__(self, orbits='EqualArmlength', **kwargs):
super().__init__(**kwargs)
def __init__(self, orbits='EqualArmlength', *args, **kwargs):
super().__init__(*args, **kwargs)
# orbits properties
self.orbits = orbits
self.orbits_start_time = None
Expand Down Expand Up @@ -577,10 +577,10 @@ class _FLR_detector(AbsSpaceDet):
and TDI. See self.orbits_init for accepted inputs. Default
'EqualArmlength'.
"""
def __init__(self, orbits='EqualArmlength', use_gpu=False, **kwargs):
def __init__(self, orbits='EqualArmlength', use_gpu=False, *args, **kwargs):
logging.warning('WARNING: FastLISAResponse TDI implementation is a work in progress. ' +
'Currently unable to reproduce LDC or BBHx waveforms.')
super().__init__(**kwargs)
super().__init__(*args, **kwargs)
self.use_gpu = use_gpu

# orbits properties
Expand Down Expand Up @@ -851,21 +851,20 @@ class space_detector(AbsSpaceDet):
The backend architecture to use for generating TDI. Accepts 'LDC'
or 'FLR'. Default 'LDC'.
"""
def __init__(self, detector='LDC', **kwargs):
super().__init__(**kwargs)
def __init__(self, detector='LDC', *args, **kwargs):
if detector == 'LDC':
self.backend = _LDC_detector(**kwargs)
self.backend = _LDC_detector(*args, **kwargs)
elif detector == 'FLR':
self.backend = _FLR_detector(**kwargs)
self.backend = _FLR_detector(*args, **kwargs)
else:
raise NotImplementedError('Unrecognized backend argument. ' +
'Currently accepts: "LDC", "FLR"')

def orbits_init(self, **kwargs):
return self.backend.orbits_init(**kwargs)
def orbits_init(self, *args, **kwargs):
return self.backend.orbits_init(*args, **kwargs)

def get_links(self, hp, hc, lamb, beta, **kwargs):
return self.backend.get_links(hp, hc, lamb, beta, **kwargs)
def get_links(self, hp, hc, lamb, beta, *args, **kwargs):
return self.backend.get_links(hp, hc, lamb, beta, *args, **kwargs)

def project_wave(self, hp, hc, lamb, beta, **kwargs):
return self.backend.project_wave(hp, hc, lamb, beta, **kwargs)
def project_wave(self, hp, hc, lamb, beta, *args, **kwargs):
return self.backend.project_wave(hp, hc, lamb, beta, *args, **kwargs)

0 comments on commit 8b40470

Please sign in to comment.