diff --git a/draco/analysis/delay.py b/draco/analysis/delay.py index e22535b96..dbbc53f0a 100644 --- a/draco/analysis/delay.py +++ b/draco/analysis/delay.py @@ -439,7 +439,7 @@ def process(self, ss): initial_S = np.ones_like(delays) * 1e1 - # Get the random Generator that we will use + # Initialize the random number generator we'll use rng = self.rng # Iterate over all baselines and use the Gibbs sampler to estimate the spectrum @@ -667,6 +667,9 @@ def process(self, ss: FreqContainerType) -> containers.DelaySpectrum: initial_S = np.ones_like(delays) * 1e1 + # Initialize the random number generator we'll use + rng = self.rng + # Iterate over all baselines and use the Gibbs sampler to estimate the spectrum for lbi, bi in delay_spec.spectrum[:].enumerate(axis=0): @@ -706,7 +709,7 @@ def process(self, ss: FreqContainerType) -> containers.DelaySpectrum: window=self.window if self.apply_window else None, fsel=non_zero_channel, niter=self.nsamp, - rng=self.rng, + rng=rng, complex_timedomain=self.complex_timedomain, )