Set upper/lower as ceiling/floor

Author: robjmcgibbon

velociraptor/autoplotter/lines.py

@@ -254,15 +254,13 @@ def create_line(
 
         if self.lower is not None:
             self.lower.convert_to_units(y.units)
-            mask = masked_y > self.lower
-            masked_x = masked_x[mask]
-            masked_y = masked_y[mask]
+            mask = masked_y < self.lower
+            masked_y[mask] = self.lower
 
         if self.upper is not None:
             self.upper.convert_to_units(y.units)
-            mask = masked_y < self.upper
-            masked_x = masked_x[mask]
-            masked_y = masked_y[mask]
+            mask = masked_y > self.upper
+            masked_y[mask] = self.upper
 
         if self.median:
             self.output = lines.binned_median_line(