Added option for plotting filled contours (or not).

aymgal · Aug 3, 2023 · 14dadf2 · 14dadf2
1 parent 0f500dc
commit 14dadf2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/coolest/api/plotting.py b/coolest/api/plotting.py
@@ -463,7 +463,7 @@ def plot_lens(self,idx_file=0):
 
 
 
-def plot_corner(parameter_id_list,chain_objs,chain_dirs,chain_names=None,point_estimate_objs=None,point_estimate_dirs=None,point_estimate_names=None,colors=None,labels=None,subplot_size=1,mc_samples_kwargs=None):
+def plot_corner(parameter_id_list,chain_objs,chain_dirs,chain_names=None,point_estimate_objs=None,point_estimate_dirs=None,point_estimate_names=None,colors=None,labels=None,subplot_size=1,mc_samples_kwargs=None,filled_contours=True):
     """
     Adding this as just a function for the moment.
     Takes a list of COOLEST files as input, which must have a chain file associated to them, and returns a corner plot.
@@ -574,7 +574,7 @@ def plot_corner(parameter_id_list,chain_objs,chain_dirs,chain_names=None,point_e
     image.triangle_plot(mcsamples,
                         params=parameter_id_list,
                         legend_labels=chain_names,
-                        filled=True,
+                        filled=filled_contours,
                         colors=colors,
                         line_args=[{'ls':'-', 'lw': 2, 'color': c} for c in colors], 
                         contour_colors=colors)