Added title to plot formatting.

choderalab · Nov 30, 2024 · 23834f5 · 23834f5
1 parent 5fd3b36
commit 23834f5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/fluorescence_assay/plotting.py b/fluorescence_assay/plotting.py
@@ -24,12 +24,16 @@ def load_data(self, Measurement: Measurements, *args, **kwargs) -> None:
     def format_plot(
         self,
         axes: matplotlib.axes._axes.Axes,
+        title: Optional[str] = None,
         xlim: Optional[list[float]] = None,
         ylim: Optional[list[float]] = None,
         xlabel: Optional[str] = None,
         ylabel: Optional[str] = None,
         square: Optional[bool] = False,
     ):
+
+        if title is not None:
+            axes.set_title(title)
 
         if xlim is not None:
             axes.set_xlim(xlim)