Fix lightcurve axis limits for forced fits (#257)

Co-authored-by: Adam Stewart <[email protected]>
askap-vast · Apr 12, 2021 · 104041d · 104041d
1 parent 249a678
commit 104041d
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/vasttools/source.py b/vasttools/source.py
@@ -620,10 +620,18 @@ def plot_lightcurve(self, sigma_thresh=5, figsize=(8, 4),
                 label=label)
 
         if yaxis_start == "0":
-            max_y = np.nanmax(
-                detections[flux_col].tolist() +
-                (sigma_thresh * upper_lims[err_value_col]).tolist()
-            )
+            if use_forced_for_limits or self.pipeline:
+                max_y = np.nanmax(
+                    detections[flux_col].tolist() +
+                    upper_lims[value_col].tolist()
+                )
+            elif use_forced_for_all:
+                max_y = np.nanmax(detections[flux_col].tolist())
+            else:
+                max_y = np.nanmax(
+                    detections[flux_col].tolist() +
+                    (sigma_thresh * upper_lims[err_value_col]).tolist()
+                )
             ax.set_ylim(
                 bottom=0,
                 top=max_y * 1.1