import matplotlib.pyplot as plt
from extinction_ensemble_props.plot_param_vs_param import plot_param_vs_param

fontsize = 20
font = {"size": fontsize}
plt.rc("font", **font)
plt.rc("lines", linewidth=2)
plt.rc("axes", linewidth=2)
plt.rc("xtick.major", width=2)
plt.rc("ytick.major", width=2)

fsize = (12, 12)
fig, ax = plt.subplots(nrows=2, ncols=2, figsize=fsize)

datasets = ["GCC09", "G03_lmc", "G24_smc"]
plot_param_vs_param(ax[0, 0], datasets, "C1", "C2")
plot_param_vs_param(ax[0, 1], datasets, "C2", "B3")
plot_param_vs_param(ax[1, 0], datasets, "C2", "C4")
plot_param_vs_param(ax[1, 1], datasets, "C4", "B3")

ax[0, 1].get_legend().remove()
ax[1, 0].get_legend().remove()
ax[1, 1].get_legend().remove()

fig.tight_layout()
plt.show()