Update pycbc_plot_kde_vals (gwastro#4445)

acorreia61201 · Apr 4, 2024 · 7e2866f · 7e2866f
1 parent 8b7e933
commit 7e2866f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bin/all_sky_search/pycbc_plot_kde_vals b/bin/all_sky_search/pycbc_plot_kde_vals
@@ -31,9 +31,9 @@ if len(args.param) != len(args.log_axis):
 
 if args.signal_file:
     signal_data = h5py.File(args.signal_file, 'r')
-    signal_kde = signal_data['signal_kde'][:]
+    signal_kde = signal_data['data_kde'][:]
 template_data = h5py.File(args.template_file, 'r')
-template_kde = template_data['template_kde'][:]
+template_kde = template_data['data_kde'][:]
 param0 = template_data[args.param[0]][:]
 if len(args.param) > 1:
     param1 = template_data[args.param[1]][:]
@@ -63,10 +63,10 @@ if args.plot_type == 'kde_vs_param':
 elif args.plot_type == 'param_vs_param':
     fig, ax = plt.subplots(1, figsize=(12,7), constrained_layout=True)
     if args.which_kde == 'signal_kde':
-        im = ax.scatter(param0, param1, marker=".", c=signal_kde, cmap='RdBu_r',
+        im = ax.scatter(param0, param1, marker=".", c=signal_kde, cmap='plasma',
                         s=5, norm=LogNorm())
     elif args.which_kde == 'template_kde':
-        im = ax.scatter(param0, param1, marker=".", c=template_kde, cmap='RdBu_r',
+        im = ax.scatter(param0, param1, marker=".", c=template_kde, cmap='plasma',
                         s=5, norm=LogNorm())
     elif args.which_kde == 'ratio_kde':
         im = ax.scatter(param0, param1, marker=".", c=signal_kde / template_kde,