adding legend for filters in early prediction

supernnova · Nov 15, 2019 · 7610d0f · 7610d0f
1 parent 364e786
commit 7610d0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supernnova/visualization/early_prediction.py b/supernnova/visualization/early_prediction.py
@@ -94,7 +94,7 @@ def plot_predictions(
                 label=f"Filter {flt}",
                 color=FILTER_COLORS[flt],
             )
-
+    ax.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.0)
     ax.set_ylabel("FLUXCAL")
     ylim = ax.get_ylim()