From b1badaee0f83821af86b39b65815f3af68e5b56e Mon Sep 17 00:00:00 2001 From: Akshar Sarvesh Date: Mon, 26 Aug 2024 15:43:34 -0700 Subject: [PATCH] STY: PR Suggestions --- archive_viewer/table_models/curve_model.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/archive_viewer/table_models/curve_model.py b/archive_viewer/table_models/curve_model.py index ad1ccff..b05da6e 100644 --- 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)