Fix issue with multiopti plots

monash-emu · Sep 1, 2023 · e0f547c · e0f547c
1 parent d1adfa3
commit e0f547c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/autumn/projects/sm_covid2/common_school/calibration_plots/opti_plots.py b/autumn/projects/sm_covid2/common_school/calibration_plots/opti_plots.py
@@ -189,9 +189,7 @@ def plot_multiple_model_fits(bcm, params_list, outfile=None):
     death_ax, rp_ax, sero_ax = axs[0], axs[1], axs[2]
 
     # set up the three axes
-    death_ax.set_ylabel("COVID-19 deaths")
-    targets["infection_deaths_ma7"].plot(style='.', ax=death_ax, label="", zorder=20, color='black')
-
+    death_ax.set_ylabel("COVID-19 deaths")   
     rp_ax.set_ylabel("Random process")
 
     if "prop_ever_infected_age_matched" in targets:
@@ -228,6 +226,8 @@ def plot_multiple_model_fits(bcm, params_list, outfile=None):
             color=colors[i]
         )
 
+    targets["infection_deaths_ma7"].plot(style='.', ax=death_ax, label="", zorder=20, color='black')
+
     # Post plotting processes
     # death
     death_ax.legend(loc='best', ncols=2)