Skip to content

Commit

Permalink
Merge pull request #325 from vallis/master
Browse files Browse the repository at this point in the history
Fix caching bug in signals.deterministic_signals.Deterministic
  • Loading branch information
vallis authored Nov 4, 2022
2 parents b4a651a + 50a12b2 commit f1ca78e
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions enterprise/signals/deterministic_signals.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def _do_selection(self, psr, waveform, selection):
sel = selection(psr)
self._keys = sorted(sel.masks.keys())
self._masks = [sel.masks[key] for key in self._keys]
self._delay = np.zeros(len(psr.toas))
self._ntoas = len(psr.toas)
self._wf, self._params = {}, {}
for key, mask in zip(self._keys, self._masks):
pnames = [psr.name, name, key]
Expand All @@ -45,9 +45,10 @@ def delay_params(self):
@signal_base.cache_call("delay_params")
def get_delay(self, params):
"""Return signal delay."""
delay = np.zeros(self._ntoas)
for key, mask in zip(self._keys, self._masks):
self._delay[mask] = self._wf[key](params=params, mask=mask)
return self._delay
delay[mask] = self._wf[key](params=params, mask=mask)
return delay

return Deterministic

Expand Down

0 comments on commit f1ca78e

Please sign in to comment.