formatting

demo_wstack_realdata.py

@@ -66,11 +66,9 @@ wgt = np.repeat(wgt[:, None], len(freq), axis=1)
 wgt[flags] = 0
 
 # flip visibilities with w<0
-mask = uvw[:,2]<0
-uvw[mask]*=-1
-uvw[mask]*=-1
-uvw[mask]*=-1
-vis = np.conj(vis)*mask.reshape(-1,1) + vis*(1-mask.reshape(-1,1))
+mask = uvw[:, 2] < 0
+uvw[mask] *= -1
+vis = np.conj(vis)*mask.reshape(-1, 1) + vis*(1-mask.reshape(-1, 1))
 
 npixdirty = 756
 DEG2RAD = np.pi/180
@@ -79,15 +77,15 @@ nthreads = 4
 epsilon = 6e-6
 t0 = time()
 print('Start gridding...')
-if (epsilon>5e-6):
+if (epsilon > 5e-6):
     dirty = ng.full_gridding_f(
         uvw.astype("f4"), freq.astype("f4"), vis.astype("c8"),
         wgt.astype("f4"), npixdirty, npixdirty, pixsize, pixsize, epsilon,
-        nthreads,verbosity=2)
+        nthreads, verbosity=2)
 else:
     dirty = ng.full_gridding(
         uvw, freq, vis, wgt, npixdirty, npixdirty, pixsize, pixsize, epsilon,
-        nthreads,verbosity=2)
+        nthreads, verbosity=2)
 print('Done')
 t = time() - t0
 print("{} s".format(t))