comment visualization and up gridder verbosity

ratt-ru · Nov 20, 2023 · b482e41 · b482e41
1 parent 2988084
commit b482e41
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/pfb/operators/gridder.py b/pfb/operators/gridder.py
@@ -609,7 +609,7 @@ def image_data_products(uvw,
         wcount = mask.sum()
         if wcount:
             ovar = ressq.sum()/wcount
-            wgt = (dof + 1)/(dof + ressq/ovar)/ovar
+            wgt = (l2reweight_dof + 1)/(l2reweight_dof + ressq/ovar)/ovar
 
     # we usually want to re-evaluate this since the robustness may change
     if robustness is not None:
@@ -647,7 +647,8 @@ def image_data_products(uvw,
         divide_by_n=False,  # hardcoded for now
         nthreads=nthreads,
         sigma_min=1.1, sigma_max=3.0,
-        double_precision_accumulation=double_accum)
+        double_precision_accumulation=double_accum,
+        verbosity=2)
     out_dict['DIRTY'] = dirty
 
     if do_psf:

diff --git a/pfb/utils/misc.py b/pfb/utils/misc.py
@@ -914,6 +914,17 @@ def concat_chan(xds, nband_out=1):
 
             out_dict = blocker.get_dask_outputs()
 
+            # import dask
+            # dask.visualize(out_dict, color="order", cmap="autumn",
+            #             node_attr={"penwidth": "4"},
+            #             filename='/home/landman/testing/pfb/out/outdict_ordered_graph.pdf',
+            #             optimize_graph=False,
+            #             engine='cytoscape')
+            # dask.visualize(out_dict,
+            #             filename='/home/landman/testing/pfb/out/outdict_graph.pdf',
+            #             optimize_graph=False, engine='cytoscape')
+            # quit()
+
             # get weighted sum of beam
             beam = sum_beam(xdst)
             l_beam = xdst[0].l_beam.data