Add optional arguments for legend locations for ECDF plots

Author: jerrymhuang

bayesflow/diagnostics/plots/calibration_ecdf.py

@@ -19,6 +19,7 @@ def calibration_ecdf(
  figsize: Sequence[float] = None,
  label_fontsize: int = 16,
  legend_fontsize: int = 14,
+ legend_location: str = "upper right",
  title_fontsize: int = 18,
  tick_fontsize: int = 12,
  rank_ecdf_color: str = "#132a70",
@@ -184,7 +185,7 @@ def calibration_ecdf(
 
  for ax, title in zip(plot_data["axes"].flat, titles):
  ax.fill_between(z, L, U, color=fill_color, alpha=0.2, label=rf"{int((1 - alpha) * 100)}$\%$ Confidence Bands")
- ax.legend(fontsize=legend_fontsize)
+ ax.legend(fontsize=legend_fontsize, loc=legend_location)
  ax.set_title(title, fontsize=title_fontsize)
 
  prettify_subplots(plot_data["axes"], num_subplots=plot_data["num_variables"], tick_fontsize=tick_fontsize)

bayesflow/diagnostics/plots/calibration_ecdf_from_quantiles.py

@@ -19,6 +19,7 @@ def calibration_ecdf_from_quantiles(
  figsize: Sequence[float] = None,
  label_fontsize: int = 16,
  legend_fontsize: int = 14,
+ legend_location: str = "upper right",
  title_fontsize: int = 18,
  tick_fontsize: int = 12,
  rank_ecdf_color: str = "#132a70",
@@ -173,7 +174,7 @@ def calibration_ecdf_from_quantiles(
  alpha=0.2,
  label=rf"{int((1 - alpha) * 100)}$\%$ Confidence Bands" + "\n(pointwise)",
  )
- ax.legend(fontsize=legend_fontsize)
+ ax.legend(fontsize=legend_fontsize, loc=legend_location)
  ax.set_title(title, fontsize=title_fontsize)
 
  prettify_subplots(plot_data["axes"], num_subplots=plot_data["num_variables"], tick_fontsize=tick_fontsize)