Need to force numpy in PSD estimate

gwastro · Nov 22, 2024 · d404116 · d404116
1 parent 8f5b539
commit d404116
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pycbc/psd/estimate.py b/pycbc/psd/estimate.py
@@ -195,6 +195,7 @@ def welch(timeseries, seg_len=4096, seg_stride=2048, window='hann',
             median_bias(len(even_psds))
         psd = (odd_median + even_median) / 2
 
+    w = w.numpy()
     psd *= 2 * delta_f * seg_len / (w*w).sum()
 
     return FrequencySeries(psd, delta_f=delta_f, dtype=timeseries.dtype,