test prange over chan in weight_data

ratt-ru · Aug 8, 2023 · 8984900 · 8984900
1 parent 98ff9be
commit 8984900
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pfb/utils/stokes.py b/pfb/utils/stokes.py
@@ -268,14 +268,14 @@ def _impl(data, weight, jones, tbin_idx, tbin_counts,
         vis = np.zeros((nrow, nchan), dtype=data.dtype)
         wgt = np.zeros((nrow, nchan), dtype=data.real.dtype)
 
-        for t in prange(nt):
+        for t in range(nt):
             for row in range(tbin_idx[t],
                              tbin_idx[t] + tbin_counts[t]):
                 p = int(ant1[row])
                 q = int(ant2[row])
                 gp = jones[t, p, :, 0]
                 gq = jones[t, q, :, 0]
-                for chan in range(nchan):
+                for chan in prange(nchan):
                     wval = wgt_func(gp[chan], gq[chan],
                                     weight[row, chan])
                     if wval > 1e-6: