add configuration options for marker [no ci]

Author: vpratz

bayesflow/diagnostics/plots/pairs_quantity.py

@@ -22,6 +22,8 @@ def pairs_quantity(
     height: float = 2.5,
     cmap: str | matplotlib.colors.Colormap = "viridis",
     alpha: float = 0.9,
+    s: float = 8.0,
+    marker: str = "o",
     label: str = None,
     label_fontsize: int = 14,
     tick_fontsize: int = 12,
@@ -94,6 +96,10 @@ def pairs_quantity(
         The colormap for the plot.
     alpha       : float in [0, 1], optional, default: 0.9
         The opacity of the plot
+    s                 : float, optional, default: 8.0
+        The marker size in points**2 for the scatter plot.
+    marker            : str, optional, default: 'o'
+        The marker for the scatter plot.
     label       : str, optional, default: None
         Label for the dataset to plot.
     label_fontsize    : int, optional, default: 14
@@ -177,7 +183,17 @@ def pairs_quantity(
 
             if i == j:
                 ax = g.axes[i, j].twinx()
-                ax.scatter(targets[:, i], values[:, i], c=row_values, cmap=cmap, s=4, vmin=vmin, vmax=vmax, alpha=alpha)
+                ax.scatter(
+                    targets[:, i],
+                    values[:, i],
+                    c=row_values,
+                    cmap=cmap,
+                    s=s,
+                    marker=marker,
+                    vmin=vmin,
+                    vmax=vmax,
+                    alpha=alpha,
+                )
                 ax.spines["left"].set_visible(False)
                 ax.spines["top"].set_visible(False)
                 ax.tick_params(axis="both", which="major", labelsize=tick_fontsize)
@@ -197,10 +213,11 @@ def pairs_quantity(
                     targets[:, i],
                     c=row_values,
                     cmap=cmap,
-                    s=4,
+                    s=s,
                     vmin=vmin,
                     vmax=vmax,
                     alpha=alpha,
+                    marker=marker,
                 )
 
     def inches_to_figure(fig, values):

bayesflow/diagnostics/plots/plot_quantity.py

@@ -26,6 +26,9 @@ def plot_quantity(
     title_fontsize: int = 18,
     tick_fontsize: int = 12,
     color: str = "#132a70",
+    s: float = 25.0,
+    marker: str = "o",
+    alpha: float = 0.5,
     xlabel: str = "Ground truth",
     ylabel: str = "",
     num_col: int = None,
@@ -90,6 +93,12 @@ def plot_quantity(
         The font size of the axis ticklabels
     color             : str, optional, default: '#8f2727'
         The color for the true vs. estimated scatter points and error bars
+    s                 : float, optional, default: 25.0
+        The marker size in points**2 for the scatter plot.
+    marker            : str, optional, default: 'o'
+        The marker for the scatter plot.
+    alpha             : float, default: 0.5
+        The opacity for the scatter plot
     num_row           : int, optional, default: None
         The number of rows for the subplots. Dynamically determined if None.
     num_col           : int, optional, default: None
@@ -143,7 +152,7 @@ def plot_quantity(
         if i >= num_variables:
             break
 
-        ax.scatter(targets[:, i], values[:, i], color=color, alpha=0.5)
+        ax.scatter(targets[:, i], values[:, i], color=color, alpha=alpha, s=s, marker=marker)
 
     prettify_subplots(axes, num_subplots=num_variables, tick_fontsize=tick_fontsize)