STY: PR Suggestions

slaclab · Aug 26, 2024 · b1badae · b1badae
1 parent 1fd0c7c
commit b1badae
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/archive_viewer/table_models/curve_model.py b/archive_viewer/table_models/curve_model.py
@@ -258,9 +258,12 @@ def curve_at_index(self, index: QModelIndex) -> ArchivePlotCurveItem:
         return self._plot.curveAtIndex(index)
 
     @Slot(object)
-    def remove_curve(self, curve: BasePlotCurveItem):
+    def remove_curve(self, curve: BasePlotCurveItem) -> None:
         """Necessary specifically for when an axis is deleted
-        To properly delete all of its connected curves"""
+        To properly delete all of its connected curves
+
+        curve: BasePlotCurveItem
+            The curve we want to delete from the model"""
         ind = self._plot._curves.index(curve)
         ind = self.index(ind, 0)
         self.removeAtIndex(ind)